File size: 2,854 Bytes
ecbefb3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 |
---
license: mit
tags:
- recommendation-system
- collaborative-filtering
- matrix-factorization
- movie-recommendations
- movielens
- machine-learning
library_name: scikit-learn
---
# DataSynthis_ML_JobTask
A powerful movie recommendation system using collaborative filtering and matrix factorization techniques on the MovieLens 100k dataset.
## Model Description
This model provides personalized movie recommendations using two state-of-the-art algorithms:
- **Collaborative Filtering (CF)**: Item-based similarity using cosine similarity
- **Matrix Factorization (SVD)**: Singular Value Decomposition for dimensionality reduction
## Dataset
- **MovieLens 100k**: 100,000 ratings from 943 users on 1,682 movies
- **User ID Range**: 1-943
- **Movie Count**: 1,682 unique movies
- **Rating Scale**: 1-5 stars
## Usage
### Python
```python
from model import predict
# Get recommendations using SVD (default)
recommendations = predict(user_id=1, n_recommendations=10, method="svd")
# Get recommendations using collaborative filtering
recommendations = predict(user_id=1, n_recommendations=10, method="cf")
print(recommendations)
```
### Parameters
- **user_id** (int): User ID between 1-943 (required)
- **n_recommendations** (int): Number of recommendations between 1-20 (default: 10)
- **method** (str): "svd" for matrix factorization or "cf" for collaborative filtering (default: "svd")
### Output
Returns a list of dictionaries with movie recommendations:
```json
[
{
"movie_id": 50,
"title": "Star Wars (1977)",
"predicted_rating": 4.5
},
{
"movie_id": 181,
"title": "Return of the Jedi (1983)",
"predicted_rating": 4.3
}
]
```
## Model Performance
- **SVD Method**: Fast predictions with good accuracy using 20 components
- **Collaborative Filtering**: More interpretable, based on item similarity
- **Cold Start Handling**: Graceful error handling for unknown users
## Technical Details
- **Framework**: Scikit-learn
- **Algorithms**: TruncatedSVD, Cosine Similarity
- **Data Processing**: Pandas for efficient matrix operations
- **Memory Efficient**: Optimized for large-scale recommendation tasks
## Installation
```bash
pip install pandas numpy scikit-learn
```
## Training
The model is pre-trained on the MovieLens 100k dataset. To retrain:
```python
from model import MovieRecommender
model = MovieRecommender()
model.load_data()
model.train()
model.save_model("movie_recommender.pkl")
```
## Citation
```bibtex
@misc{datasynthis_ml_jobtask,
title={DataSynthis ML JobTask: Movie Recommendation System},
author={tasdid25},
year={2025},
url={https://huggingface.co/tasdid25/DataSynthis_ML_JobTask}
}
```
## License
MIT License - see LICENSE file for details. |