Arpan-Mishra / Movie-Genre-Prediction

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

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


Languages

Language:Jupyter Notebook 100.0%Language:Python 0.0%