Performance Analysis of Regression Models on Predicting Movie Ratings

This research explores the application and performance comparison of various machine learning for movie ratings. This study aims to make a comprehensive comparison in performance between various machine learning models such as XGBoost, Random Forest Regressor, K Nearest Neighbor Regression, Support Vector Regressor and Linear Regression. The dataset used is MovieLens 20M which includes a rating dataset and a movie dataset. We employ a content-based recommendation system in conjunction to applying machine learning models to predict movie ratings based on user meta data. Our methodology involves splitting the data into 70:30 train-test data and evaluating them using Mean Absolute Error (MAE), Mean Square Error (MSE), and Root Mean Squared Error (RMSE). The findings of our research conclude that XGBoost is the superior model which is shown from the RMSE value of XGBoost being the lowest (0.8727).

Authors

Institutions

Publication Details

Published
2024-09-12
DOI
https://doi.org/10.1109/ictiia61827.2024.10761200
Primary Topic
Stock Market Forecasting Methods
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Performance Analysis of Regression Models on Predicting Movie Ratings

Henry Lucky, Hendry Hartono, Derwin Suhartono, Ignatius Kennard
Stock Market Forecasting Methods
article

Performance Analysis of Regression Models on Predicting Movie Ratings

Henry Lucky, Hendry Hartono, Derwin Suhartono, Ignatius Kennard
article en

Abstract

This research explores the application and performance comparison of various machine learning for movie ratings. This study aims to make a comprehensive comparison in performance between various machine learning models such as XGBoost, Random Forest Regressor, K Nearest Neighbor Regression, Support Vector Regressor and Linear Regression. The dataset used is MovieLens 20M which includes a rating dataset and a movie dataset. We employ a content-based recommendation system in conjunction to applying machine learning models to predict movie ratings based on user meta data. Our methodology involves splitting the data into 70:30 train-test data and evaluating them using Mean Absolute Error (MAE), Mean Square Error (MSE), and Root Mean Squared Error (RMSE). The findings of our research conclude that XGBoost is the superior model which is shown from the RMSE value of XGBoost being the lowest (0.8727).

Binus University (ID)
Gender equality
Openalex Percentile: Top 73%
Stock Market Forecasting Methods
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.