Upload folder using huggingface_hub
Browse files- MF/REPORT.txt +5 -0
- MF/best.weights.h5 +3 -0
- MF/eval_test_metrics.csv +2 -0
- MF/eval_test_metrics.json +7 -0
- MF/hparams.json +28 -0
- MF/metrics.json +11 -0
- MF/test_metrics.csv +2 -0
- MF/train_log.csv +22 -0
MF/REPORT.txt
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
Model: neumf | Loss: bpr
|
| 2 |
+
Experiment: exp-neumf-64-2
|
| 3 |
+
Users: 260039 | Items: 365797
|
| 4 |
+
Best val Recall@10: 0.6105 (epoch 18)
|
| 5 |
+
Test: P@10=0.0571 R@10=0.5713 F1@10=0.1039 MAP@10=0.2927 NDCG@10=0.3587
|
MF/best.weights.h5
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6912f24ca4ac21f26fc4e7bbfa14e9e888cd67dd0d34b1cdda8ca79a69cd5642
|
| 3 |
+
size 320551344
|
MF/eval_test_metrics.csv
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
precision,recall,f1,map,ndcg
|
| 2 |
+
0.057159887555328234,0.5715988755532824,0.10392706828241499,0.2927710797592815,0.35885029532421475
|
MF/eval_test_metrics.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"precision": 0.057159887555328234,
|
| 3 |
+
"recall": 0.5715988755532824,
|
| 4 |
+
"f1": 0.10392706828241499,
|
| 5 |
+
"map": 0.2927710797592815,
|
| 6 |
+
"ndcg": 0.35885029532421475
|
| 7 |
+
}
|
MF/hparams.json
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"data_dir": "artifacts/movies_from_music-2/data",
|
| 3 |
+
"out_root": "checkpoint",
|
| 4 |
+
"model": "neumf",
|
| 5 |
+
"loss": "bpr",
|
| 6 |
+
"exp_name": "exp-neumf-64-2",
|
| 7 |
+
"epochs": 30,
|
| 8 |
+
"batch_size": 1024,
|
| 9 |
+
"emb_dim": 64,
|
| 10 |
+
"lr": 0.003,
|
| 11 |
+
"l2": 1e-06,
|
| 12 |
+
"topk": 10,
|
| 13 |
+
"eval_negs": 99,
|
| 14 |
+
"patience": 3,
|
| 15 |
+
"seed": 42,
|
| 16 |
+
"resume": false,
|
| 17 |
+
"save_embeddings": false,
|
| 18 |
+
"itemknn_topk_neighbors": 200,
|
| 19 |
+
"lightgcn_layers": 3,
|
| 20 |
+
"cache_lightgcn": true,
|
| 21 |
+
"policy_name": "none",
|
| 22 |
+
"epsilon": 0.1,
|
| 23 |
+
"alpha": 1.0,
|
| 24 |
+
"replay_after_train": false,
|
| 25 |
+
"candidate_topn": 200,
|
| 26 |
+
"slate_k": 10,
|
| 27 |
+
"bootstrap_iters": 1000
|
| 28 |
+
}
|
MF/metrics.json
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_val_recall": 0.6105204422637207,
|
| 3 |
+
"best_epoch": 18,
|
| 4 |
+
"test": {
|
| 5 |
+
"precision": 0.057131045727756215,
|
| 6 |
+
"recall": 0.5713104572775622,
|
| 7 |
+
"f1": 0.10387462859592039,
|
| 8 |
+
"map": 0.29267769633089835,
|
| 9 |
+
"ndcg": 0.35869102500499395
|
| 10 |
+
}
|
| 11 |
+
}
|
MF/test_metrics.csv
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
model,loss,exp_name,timestamp,best_epoch,best_val_recall,test_precision,test_recall,test_f1,test_map,test_ndcg,epochs_run,emb_dim,lr,l2,batch_size,topk,eval_negs,seed,itemknn_topk_neighbors,lightgcn_layers,policy_name,epsilon,alpha,candidate_topn,slate_k,bootstrap_iters
|
| 2 |
+
neumf,bpr,exp-neumf-64-2,2025-09-02T20:54:12,18,0.6105204422637207,0.057131045727756215,0.5713104572775622,0.10387462859592039,0.29267769633089835,0.35869102500499395,21,64,0.003,1e-06,1024,10,99,42,200,3,none,0.1,1.0,200,10,1000
|
MF/train_log.csv
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
epoch,train_loss,val_precision,val_recall,val_f1,val_map,val_ndcg,is_best
|
| 2 |
+
1,0.4332443844435111,0.05776175255793643,0.5776175255793643,0.10502136828715714,0.2970680904332249,0.363401807198025,1
|
| 3 |
+
2,0.4221300712535081,0.05839373116042673,0.5839373116042673,0.106170420291685,0.2998897219192102,0.36700822138137806,1
|
| 4 |
+
3,0.3981855443050257,0.058811006859575805,0.5881100685957581,0.10692910338104693,0.29974338083618224,0.36795376431695764,1
|
| 5 |
+
4,0.37657057752645445,0.058739439225128726,0.5873943922512873,0.10679898040932499,0.30108399598859265,0.36875952099624354,0
|
| 6 |
+
5,0.36171849572454956,0.05969711121489091,0.5969711121489092,0.10854020220889256,0.31136125125717495,0.37889346579433286,1
|
| 7 |
+
6,0.35156996092941095,0.05996458008600245,0.5996458008600246,0.10902650924727722,0.31264725685801137,0.3805723807582111,1
|
| 8 |
+
7,0.34525731386903113,0.06003008266668283,0.6003008266668284,0.10914560484851427,0.3132384094656427,0.38124417458153415,1
|
| 9 |
+
8,0.3394609433725467,0.05970863481704765,0.5970863481704766,0.10856115421281395,0.30804575592086436,0.37645880778928925,0
|
| 10 |
+
9,0.3338334590554765,0.06006101444089301,0.6006101444089302,0.10920184443798732,0.31268827058072496,0.3809059433293145,1
|
| 11 |
+
10,0.3302042118399148,0.059904536053712105,0.5990453605371212,0.10891733827947661,0.31250389294621733,0.38036872594166926,0
|
| 12 |
+
11,0.3269095296613933,0.06018231551622706,0.6018231551622705,0.10942239184768557,0.3135066990444267,0.3817811479869565,1
|
| 13 |
+
12,0.32374349191247626,0.05958672723633694,0.5958672723633694,0.10833950406606721,0.3107562645507138,0.37830376161550133,0
|
| 14 |
+
13,0.32078941612210476,0.06073241589286689,0.6073241589286689,0.11042257435066709,0.31859706256930886,0.38700754024999534,1
|
| 15 |
+
14,0.31883900507502283,0.060627490462702945,0.6062749046270295,0.11023180084127812,0.31987454099336143,0.3877318944143875,0
|
| 16 |
+
15,0.31878021661048445,0.06078457535526053,0.6078457535526053,0.11051740973683731,0.31967137131924644,0.38791024493134885,1
|
| 17 |
+
16,0.3163065833995645,0.06041885261312841,0.6041885261312842,0.10985245929659712,0.3181031000415216,0.3858787978054781,0
|
| 18 |
+
17,0.314131417582112,0.06051649997877232,0.6051649997877231,0.11002999996140425,0.3188828686009953,0.38673444060865514,0
|
| 19 |
+
18,0.312700189758448,0.061052044226372074,0.6105204422637207,0.11100371677522197,0.32074223839184807,0.38942340723714847,1
|
| 20 |
+
19,0.3108912939930324,0.06092164557038796,0.6092164557038798,0.11076662830979635,0.31984155624658656,0.3883655961784558,0
|
| 21 |
+
20,0.31016012417372835,0.06073241589286689,0.6073241589286689,0.11042257435066709,0.31813274900872474,0.3866863852633505,0
|
| 22 |
+
21,0.30859343544506407,0.06088343573165776,0.6088343573165776,0.11069715587574143,0.31432612150305117,0.384078808150901,0
|