Skip to content

This project deals with Genre prediction based on the description of a movie. It is trained on a movie dataset containing 40000+ movies along with their genres, description, production companies and many other fields.

Notifications You must be signed in to change notification settings

adithya1919/Text-Classification_MadeWithML

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 

Repository files navigation

Text-Classification_MadeWithML

This project deals with Genre prediction based on the description of a movie. It is trained on a movie dataset containing 40000+ movies along with their genres, description, production companies and many other fields. We used a Logistic Regression model for genre prediction. As 'description' is the most important field from which we can predict the genre, we used that field. We varied the Test data set size from 0.05 to o.40 and number of iterations from 10 to 50. We got highest F1 score of 0.4635 at test data size=0.1 and number of iterations =30.

About

This project deals with Genre prediction based on the description of a movie. It is trained on a movie dataset containing 40000+ movies along with their genres, description, production companies and many other fields.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published