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
- Henry Lucky (ORCID: https://orcid.org/0000-0002-4233-0409)
- Hendry Hartono (ORCID: https://orcid.org/0000-0001-6489-4929)
- Derwin Suhartono (ORCID: https://orcid.org/0000-0002-3271-5874)
- Ignatius Kennard
Institutions
- Binus University (ID)
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