farchan commited on
Commit
57db31a
·
verified ·
1 Parent(s): 0c24587

Upload folder using huggingface_hub

Browse files
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