Skip to content

A tool to predict the IMDB rating of a movie using machine learning

License

Notifications You must be signed in to change notification settings

robjweiss/Movie-Data-Analytics

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

52 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Movie-Data-Analytics

A tool to predict the IMDB rating of a movie using machine learning

Features:

  • Predict the IMDB rating of any movie
  • Calculate error of predictions
  • Display Movie data and poster from IMDB
  • Interactive GUI

Setup

Dependencies

  • tkinter (The main Tk software will also be needed if not installed)
  • Pillow
  • pandas
  • scipy
  • scikit-learn

Installation

git clone https://github.com/robjweiss/Movie-Data-Analytics.git

Instructions

  1. Change directories to the src folder using cd src
  2. Launch the program by running the main file using python main.py
  3. Enter a move title and the year it was made, if known

Usage

The following example uses a movie where the year is specified and a movie without a year being specified: Alt text

Example result using Star Wars (2017): Alt text

Additional Information

  • The system implements machine learning using random forests and predicts the rating of a given movie based on data from over 10,000 films
  • This program is written in python 3 and uses Tk as a user interface
  • The OMDb API was used to obtain the IMDb entries
  • The data that the machine learning model is based on was obtained from this dataset

About

A tool to predict the IMDB rating of a movie using machine learning

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages