Skip to content

feat: Improve TF-IDF weighting, match explanations, and MovieLens enrichment #5

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
LavishSphere merged 4 commits into from
Jun 22, 2026
Merged

feat: Improve TF-IDF weighting, match explanations, and MovieLens enrichment #5

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
64f9cd6
feat: Improve TF-IDF weighting and expose match breakdown
LavishSphere Jun 20, 2026
9c6bf60
feat: integrate MovieLens metadata into recommendations
LavishSphere Jun 22, 2026
5dfb53a
docs: clarify included MovieLens dataset wording
LavishSphere Jun 22, 2026
30f7899
style: format data loader with black
LavishSphere Jun 22, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -217,3 +217,6 @@ __marimo__/
# Streamlit
.streamlit/secrets.toml
.DS_Store

# Large local datasets
backend/data/ml-32m/ratings.csv
11 changes: 9 additions & 2 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ StreamCompass is a content-based recommendation engine that takes a show or movi
- pandas / numpy

**Data**
- [MovieLens Dataset](https://grouplens.org/datasets/movielens/)
- [MovieLens 32M Dataset](https://grouplens.org/datasets/movielens/) for movie genres, user tags, and cached aggregate ratings
- Streaming platform availability data via Kaggle

---
Expand All @@ -51,6 +51,10 @@ uvicorn main:app --reload

The API will be running at `http://localhost:8000`. Docs available at `http://localhost:8000/docs`.

### MovieLens Data

StreamCompass includes MovieLens 32M metadata under `backend/data/ml-32m/` for movie genres, user tags, and cached aggregate rating signals. The app automatically uses `movies.csv`, `tags.csv`, `links.csv`, and `ratings_summary.csv` during backend startup.

### Frontend Setup

```bash
Expand All @@ -70,6 +74,9 @@ The app will be running at `http://localhost:5173`.
| GET | `/search?q={title}` | Search for a title |
| POST | `/recommend` | Get recommendations for a given title |
| GET | `/platforms` | List all supported streaming platforms |
| GET | `/genres` | List available genres |
| GET | `/title/{title}` | Get metadata for a title |
| GET | `/health` | Check API/data load status |

---

Expand Down Expand Up @@ -117,4 +124,4 @@ StreamCompass/
## Course

CS 4100 - Artificial Intelligence
Northeastern University, Summer 2026
Northeastern University, Summer 2026
4 changes: 4 additions & 0 deletions backend/data/ml-32m/checksums.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
8f033867bcb4e6be8792b21468b4fa6e links.csv
0df90835c19151f9d819d0822e190797 movies.csv
cf12b74f9ad4b94a011f079e26d4270a ratings.csv
963bf4fa4de6b8901868fddd3eb54567 tags.csv
Loading
Loading