swirl commited on
Commit
b7e1a83
·
verified ·
1 Parent(s): aab43a9

Upload model_mf_v3 (epoch 5, 75u/194w, RMSE=1.1956) (model card)

Browse files
Files changed (1) hide show
  1. README.md +8 -8
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 | 64 |
22
  | Side features | Yes (768-dim wine embeddings) |
23
- | Parameters | 123,654 |
24
- | Users | 73 |
25
- | Wines | 188 |
26
- | Best epoch | 7 |
27
- | Val RMSE | 0.8635 |
28
- | Val MAE | 0.606194019317627 |
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("model_mf_v2.pt")
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