Upload model_mf_v3 (epoch 5, 75u/194w, RMSE=1.1956) (model card)
Browse files
README.md
CHANGED
|
@@ -18,14 +18,14 @@ real user-wine ratings from the Swirl platform.
|
|
| 18 |
| Property | Value |
|
| 19 |
|---|---|
|
| 20 |
| Architecture | Biased MF with optional side-feature projection |
|
| 21 |
-
| Latent dimension |
|
| 22 |
| Side features | Yes (768-dim wine embeddings) |
|
| 23 |
-
| Parameters |
|
| 24 |
-
| Users |
|
| 25 |
-
| Wines |
|
| 26 |
-
| Best epoch |
|
| 27 |
-
| Val RMSE |
|
| 28 |
-
| Val MAE | 0.
|
| 29 |
| Version | mf-v1 |
|
| 30 |
|
| 31 |
## Scoring formula
|
|
@@ -44,7 +44,7 @@ Two-stage scoring for production:
|
|
| 44 |
```python
|
| 45 |
from matrix_factorization.inference import MFScorer
|
| 46 |
|
| 47 |
-
scorer = MFScorer.load("
|
| 48 |
score = scorer.score(user_id="...", wine_id="...")
|
| 49 |
```
|
| 50 |
|
|
|
|
| 18 |
| Property | Value |
|
| 19 |
|---|---|
|
| 20 |
| Architecture | Biased MF with optional side-feature projection |
|
| 21 |
+
| Latent dimension | 16 |
|
| 22 |
| Side features | Yes (768-dim wine embeddings) |
|
| 23 |
+
| Parameters | 29,710 |
|
| 24 |
+
| Users | 75 |
|
| 25 |
+
| Wines | 194 |
|
| 26 |
+
| Best epoch | 5 |
|
| 27 |
+
| Val RMSE | 1.1956 |
|
| 28 |
+
| Val MAE | 0.7387912273406982 |
|
| 29 |
| Version | mf-v1 |
|
| 30 |
|
| 31 |
## Scoring formula
|
|
|
|
| 44 |
```python
|
| 45 |
from matrix_factorization.inference import MFScorer
|
| 46 |
|
| 47 |
+
scorer = MFScorer.load("model_mf_v3.pt")
|
| 48 |
score = scorer.score(user_id="...", wine_id="...")
|
| 49 |
```
|
| 50 |
|