Powered by OpenAIRE graph
Found an issue? Give us feedback
image/svg+xml art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos Open Access logo, converted into svg, designed by PLoS. This version with transparent background. http://commons.wikimedia.org/wiki/File:Open_Access_logo_PLoS_white.svg art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos http://www.plos.org/ ZENODOarrow_drop_down
image/svg+xml art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos Open Access logo, converted into svg, designed by PLoS. This version with transparent background. http://commons.wikimedia.org/wiki/File:Open_Access_logo_PLoS_white.svg art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos http://www.plos.org/
ZENODO
Article . 2023
License: CC BY
Data sources: ZENODO
ZENODO
Article . 2023
License: CC BY
Data sources: Datacite
ZENODO
Article . 2023
License: CC BY
Data sources: Datacite
versions View all 2 versions
addClaim

Effective Use of Content as a Feature in IMDB Dataset Analysis

Authors: Yasemin Atayolu*; Yakup Kutlu;

Effective Use of Content as a Feature in IMDB Dataset Analysis

Abstract

This study uses data from IMDb and TMDB to build two machine learning models. One model predicts movie ratings, and the other classifies movie genres. To predict ratings, we work with a carefully selected subset of IMDb data, including details like titles, genres, ratings, and crew roles, to create a structured dataset. We focus on Gradient Boosting Decision Trees (GBDT), including XGBoost and CatBoost models, and check their performance with metrics like Mean Squared Error (MSE) and Mean Absolute Error (MAE). Out of the tested models, the Gradient Boosting Regressor gives the best results, achieving a balance between speed and accuracy. For genre classification, we collect plot summaries from TMDB. Using Sentence Transformers, we create embeddings that capture the relationships between genres. We feed these embeddings into a convolutional neural network (CNN) that incorporates Conv2D layers with a 1x2 kernel size and MaxPooling2D with a 1x2 pool size. Results suggest that content management may optimize rating and genre prediction and provide insights.

Keywords

Genre classification, Machine learning, Sentence Transformers, Movie rating prediction, CNN

  • BIP!
    Impact byBIP!
    selected citations
    These citations are derived from selected sources.
    This is an alternative to the "Influence" indicator, which also reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
    0
    popularity
    This indicator reflects the "current" impact/attention (the "hype") of an article in the research community at large, based on the underlying citation network.
    Average
    influence
    This indicator reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
    Average
    impulse
    This indicator reflects the initial momentum of an article directly after its publication, based on the underlying citation network.
    Average
Powered by OpenAIRE graph
Found an issue? Give us feedback
selected citations
These citations are derived from selected sources.
This is an alternative to the "Influence" indicator, which also reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
BIP!Citations provided by BIP!
popularity
This indicator reflects the "current" impact/attention (the "hype") of an article in the research community at large, based on the underlying citation network.
BIP!Popularity provided by BIP!
influence
This indicator reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
BIP!Influence provided by BIP!
impulse
This indicator reflects the initial momentum of an article directly after its publication, based on the underlying citation network.
BIP!Impulse provided by BIP!
0
Average
Average
Average
Green