xuwd11 / movie_classification

Deep Learning for Movie Genre Classification (RNN; CNN; combined network)

Deep Learning for Movie Genre Classification

In this project, we implemented several machine learning models to predict the genres of movies using their overview descriptions and posters (since we want to predict genres for new movies, we didn't use features like ratings, which are not available for new movies). We further compared those models and made a machine learning pipeline for the best model. The best model is a concatenated model composed of an encoder for posters and an encoder for overview texts.

Our pipeline can be used to automatically tag the genres of movies on movie-hosting websites like Netflix and movie-database websites like TMDB.

Please view our final report here.
Please view our poster here.

movie_classification's Issues

Deep learning

Baseline models

Final Report

https://www.overleaf.com/15696487fgqhdmrpxjrw

Data scrapping

Scrape movie genres and descriptions from:

1. TMDB (done)
2. IMDB (done)
3. Wikepedia (?)

Milestone2

At least 4-5 papers in literature review

2 paragraphs of problem statement

Around 2 pages in total

EDA

Repeat EDAs in

1. https://spandan-madan.github.io/DeepLearningProject/
2. https://kevin11h.github.io/YelpDatasetChallengeDataScienceAndMachineLearningUCSD/