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.
-
Notifications
You must be signed in to change notification settings - Fork 1
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.
adithya1919/Text-Classification_MadeWithML
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
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 0
No packages published