Movie genre preference prediction using machine learning for customer-based information

Haifeng Wang, Haili Zhang

Research output: Chapter in Book/Report/Conference proceedingConference contribution

5 Scopus citations

Abstract

This work introduces a movie genre preference predictive model usable by small and medium-sized enterprises (SMEs) who are in need of a data-based and analytical approach to stock proper movies for local audiences and retain more customers. We used classification models to extract features from one thousand customers' demographic, behavioral and social information to predict their movie genre preference. In the implementation, a Gaussian kernel support vector machine (SVM) classification model and a logistic regression model were established to extract features from sample data and their test error-in-sample were compared. Comparison of error-out-sample was also made under different Vapnik-Chervonenkis (VC) dimensions in the machine learning algorithm to find and prevent overfitting. Gaussian kernel SVM prediction model can correctly predict movie genre preferences in 85% of positive cases. The accuracy of the algorithm increased to 93% with a smaller VC dimension and less overfitting. These findings advance our understanding of how to use machine learning approach to predict customers' preferences with a small data set and design prediction tools for these enterprises.

Original languageEnglish (US)
Title of host publication2018 IEEE 8th Annual Computing and Communication Workshop and Conference, CCWC 2018
EditorsSatyajit Chakrabarti, Himadri Nath Saha
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages110-116
Number of pages7
ISBN (Electronic)9781538646496
DOIs
StatePublished - Feb 22 2018
Event8th IEEE Annual Computing and Communication Workshop and Conference, CCWC 2018 - Las Vegas, United States
Duration: Jan 8 2018Jan 10 2018

Publication series

Name2018 IEEE 8th Annual Computing and Communication Workshop and Conference, CCWC 2018
Volume2018-January

Other

Other8th IEEE Annual Computing and Communication Workshop and Conference, CCWC 2018
CountryUnited States
CityLas Vegas
Period1/8/181/10/18

All Science Journal Classification (ASJC) codes

  • Artificial Intelligence
  • Computer Networks and Communications
  • Computer Science Applications
  • Signal Processing

Fingerprint Dive into the research topics of 'Movie genre preference prediction using machine learning for customer-based information'. Together they form a unique fingerprint.

Cite this