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 K-binary transformation, rank method and probabilistic classification with learned probability threshold are employed for the multi-label 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 F-score of 0.56, and a hit rate of 80.5%.
Code (1)
Tasks
BIG-bench Machine LearningClassificationGeneral Classificationtext-classificationText ClassificationSimilar Papers 제목 키워드 기반
Prediction of a Movie's Success From Plot Summaries Using Deep Learning Models
As the size of investment for movie production grows bigger, the need for predicting a movie{'}s success in early stages has increased. To address this need, various approaches have been proposed, mostly relying on movie…
Unraveling Movie Genres through Cross-Attention Fusion of Bi-Modal Synergy of Poster
Movie posters are not just decorative; they are meticulously designed to capture the essence of a movie, such as its genre, storyline, and tone/vibe. For decades, movie posters have graced cinema walls, billboards, and n…
Genre classificationMarketingOptical Character Recognition (OCR)Recommendation SystemsA movie genre prediction based on Multivariate Bernoulli model and genre correlations
Movie ratings play an important role both in determining the likelihood of a potential viewer to watch the movie and in reflecting the current viewer satisfaction with the movie. They are available in several sources lik…
Demystifying Visual Features of Movie Posters for Multi-Label Genre Identification
In the film industry, movie posters have been an essential part of advertising and marketing for many decades, and continue to play a vital role even today in the form of digital posters through online, social media and …
MarketingNarraSum: A Large-Scale Dataset for Abstractive Narrative Summarization
Narrative summarization aims to produce a distilled version of a narrative to describe its most salient events and characters. Summarizing a narrative is challenging as it requires an understanding of event causality and…
Natural Language Understanding