/IMDBPrediction

Machine Learning project for the course 406 at CSU Long Beach

Primary LanguagePython

IMDBPrediction

Machine Learning project for the course 406 at CSU Long Beach

  • Dataset:

  • Features:

    • keywords : string X
    • adult : boolean X
    • budget : int X
    • genres : string X
    • original_language : string X
    • production_companies : string X
    • production_country : string X
    • revenue : int X
    • spoken_languages : string X
    • runtime : int X
    • popularity : int X
  • Label:

    • Vote average 0.0 to 10.0
  • output:

    • Vote average: 0.0 to 10.0
  • Algo:

    • Logistic regression
    • softmax
    • SVM