Skip to content

vuchau8592/IMDB-rating-predict

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 

Repository files navigation

IMDB_rating_predict

Extension version of previous IMDB classification project. This project tries to predict the IMDB score of movies with different regressors.

The dataset used in this project is 5000 IMDB movie. Here is the link of original dataset from dataworld: https://data.world/data-society/imdb-5000-movie-dataset.

Technique and model used:

  • Data cleaning & analysis: Numpy, Pandas
  • Data visualization: Plotly, Seaborn
  • Data preprocessing: OneHotEnconder, Feature Selection, Polynomial Feature
  • Models:
    • Linear Regressor, RANSAC, SGDRegressor
    • SVC
    • DecisionTreeRegressor
    • Ensemble: VotingRegressor, Bagging, RandomForest, Adaboost, XGBoost

About

A further approach of IMDB classification project

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published