A pipeline movie recommendation system developed as part of the Machine Learning (ML) course.
This project implements an hybrid methodology to solve the cold-start problem and improve recommendation accuracy:
A feature-based machine learning approach that treats recommendation as a regression/ranking problem:
- Content Encoding: Metadata is encoded into numerical features.
- Collaborative Features: User and Item embeddings/vectors are used as input features.
- Gradient Boosting Machine (GBM): A Tree-based model (Hist Gradient Boosting) is trained on these combined features to predict the specific user rating.
This project utilizes a custom dataset constructed by merging user ratings with rich movie metadata collected from multiple sources:
-
MovieLens Dataset
User–item ratings are derived from the MovieLens datasets provided by GroupLens Research.
These ratings form the backbone of the collaborative filtering pipeline. -
The Movie Database (TMDB) API
Rich movie metadata (TMDB IDs, cast, crew, directors, genres) was collected programmatically via the TMDB API and used primarily for content-based modeling. -
Rotten Tomatoes Reviews Dataset
A large-scale dataset of critic and audience reviews scraped from Rotten Tomatoes, used to enrich item representations and provide additional signals for cold-start movies and sentiment-aware modeling.
Due to GitHub file size limitations, the trained models and datasets are not included in this repository.
You can download the trained models and datasets from Hugging Face: https://huggingface.co/jacob14047/Fidelio/
