wsklee/cx1115-project-movieclassification

Movie Genre Prediction Model using Multi-label classification

CX1115 Group Project : Movie Genre Prediction Model using Multi-label classification

---

Project Overview

---

Project Period: AY20/21 Semester 2

Members

• Lim Qing Chuan : Data Analysis
• Kwok Zong Heng : Data Analysis
• Lee Wonseok : API data collection, machine learning

❓ Problem Definition

Are we able to predict the genre of a movie just with its keywords?

💾 Database chosen

• TMDB Movie Database API
• 28437 Movies

💻 Tech Stack

• Python 3.8.5
• Jupyter Notebook

Updates

• New classification using Scikit-multilearn
• New evaluation method - Hamming loss

Evaluation Methods

• F1 scores
• Hamming loss

Final Score

Prediction Example