Add movie_recommender model
Browse files- movie_recommender/evaluation_results.json +22 -0
- movie_recommender/item_features.pkl +3 -0
- movie_recommender/lightfm_dataset.pkl +3 -0
- movie_recommender/lightfm_model.pkl +3 -0
- movie_recommender/test_interactions.pkl +3 -0
- movie_recommender/train_interactions.pkl +3 -0
- movie_recommender/training_config.pkl +3 -0
movie_recommender/evaluation_results.json
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"metrics": {
|
| 3 |
+
"test_precision@10": 0.2617930769920349,
|
| 4 |
+
"test_recall@10": 0.1444481732669243,
|
| 5 |
+
"test_auc": 0.9946063160896301,
|
| 6 |
+
"train_precision@10": 0.489393949508667,
|
| 7 |
+
"train_recall@10": 0.09161614057386509,
|
| 8 |
+
"train_auc": 0.9965494871139526
|
| 9 |
+
},
|
| 10 |
+
"model_config": {
|
| 11 |
+
"min_rating": 3.0,
|
| 12 |
+
"test_percentage": 0.2,
|
| 13 |
+
"random_seed": 42,
|
| 14 |
+
"loss": "warp",
|
| 15 |
+
"no_components": 64,
|
| 16 |
+
"epochs": 30
|
| 17 |
+
},
|
| 18 |
+
"evaluation_notes": {
|
| 19 |
+
"masking_applied": true,
|
| 20 |
+
"description": "Test metrics computed with train_interactions masking to prevent data leakage"
|
| 21 |
+
}
|
| 22 |
+
}
|
movie_recommender/item_features.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e510de56403bc51ab2715ef58dcb70c8fb9da3439aabd6e94c98f0aadcc44b55
|
| 3 |
+
size 2522604
|
movie_recommender/lightfm_dataset.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1a90cfd41729c753eb57ffb84788225cc6518862a2c2d898d1e8a96814204b1f
|
| 3 |
+
size 8808854
|
movie_recommender/lightfm_model.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4f0ddd0d7a918b4c848ce09ac915e920f5697c84035dedc2f3929ecffda4a6f0
|
| 3 |
+
size 214100867
|
movie_recommender/test_interactions.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0330c3c4680cd2ffcbe540a1e69a7bcdbf7dc8fe0e81ae8fed9f6df06b02c26d
|
| 3 |
+
size 63080476
|
movie_recommender/train_interactions.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ae96b664ddc1f54b7d460ce889de0bfda7bbdd8a024ab7c069eb86c995ffdced
|
| 3 |
+
size 252320436
|
movie_recommender/training_config.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b17c4b016be8815c1340eb8d6369ee30471922575ff0bc134403f8426808f41a
|
| 3 |
+
size 124
|