|
|
--- |
|
|
language: en |
|
|
license: mit |
|
|
tags: |
|
|
- recommendation-system |
|
|
- collaborative-filtering |
|
|
- matrix-factorization |
|
|
- movielens |
|
|
- svd |
|
|
- internship-task |
|
|
datasets: |
|
|
- movielens |
|
|
model-index: |
|
|
- name: DataSynthis_ML_JobTask |
|
|
results: |
|
|
- task: |
|
|
type: recommendation |
|
|
name: Movie Recommendation |
|
|
dataset: |
|
|
name: MovieLens |
|
|
type: movielens |
|
|
metrics: |
|
|
- type: precision@k |
|
|
value: 0.7460454747522295 |
|
|
- type: recall@k |
|
|
value: 0.5147626084794534 |
|
|
--- |
|
|
|
|
|
# 🎬 Movie Recommendation System (DataSynthis ML Job Task) |
|
|
|
|
|
This model was built using the MovieLens dataset for the **ML Engineer Intern task**. |
|
|
|
|
|
### Features |
|
|
Item-based Collaborative Filtering |
|
|
Matrix Factorization (SVD) |
|
|
Evaluation Metrics: Precision = 0.7460454747522295, Recall = 0.5147626084794534 |
|
|
|
|
|
### How to Use |
|
|
```python |
|
|
from joblib import load |
|
|
model = load("model.joblib") |
|
|
# Use recommend_movies(user_id, N) function |
