/Movie-Genre-Prediction

Predicting movie genres using resnet 50 in a multi-label classification problem.

Primary LanguageJupyter Notebook

Movie-Genre-Prediction

Predicting movie genres from their posters using resnet 50 in a multi-label classification problem.

The model currently has an fbeta score of 0.55. This has only been trained on 1400 posters out of 39000+ available posters, and has 22 genres in total.

Data Source: https://www.kaggle.com/neha1703/movie-genre-from-its-poster