Predicting Movie Genres Based on Plot Summaries
This topic contains 0 replies, has 1 voice, and was last updated by arXiv 10 months, 4 weeks ago.

Predicting Movie Genres Based on Plot Summaries
This project explores several Machine Learning methods to predict movie genres based on plot summaries. Naive Bayes, Word2Vec+XGBoost and Recurrent Neural Networks are used for text classification, while Kbinary transformation, rank method and probabilistic classification with learned probability threshold are employed for the multilabel problem involved in the genre tagging task.Experiments with more than 250,000 movies show that employing the Gated Recurrent Units (GRU) neural networks for the probabilistic classification with learned probability threshold approach achieves the best result on the test set. The model attains a Jaccard Index of 50.0%, a Fscore of 0.56, and a hit rate of 80.5%.
Predicting Movie Genres Based on Plot Summaries
by Quan Hoang
https://arxiv.org/pdf/1801.04813v1.pdf
You must be logged in to reply to this topic.