farchan commited on
Commit
7ccd359
·
verified ·
1 Parent(s): 85e2153

Upload folder using huggingface_hub

Browse files
ITMKNN/REPORT.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ Model: itemknn | Loss: bpr
2
+ Experiment: exp-itemknn-2
3
+ Users: 260039 | Items: 365797
4
+ Best val Recall@10: 0.9396 (epoch 19)
5
+ Test: P@10=0.0956 R@10=0.9556 F1@10=0.1738 MAP@10=0.8720 NDCG@10=0.8932
ITMKNN/hparams.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_dir": "artifacts/movies_from_music-2/data",
3
+ "out_root": "checkpoint",
4
+ "model": "itemknn",
5
+ "loss": "bpr",
6
+ "exp_name": "exp-itemknn-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
+ }
ITMKNN/item_embeddings.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a43d1fa72d6a61d9c231041999f5a06e6bc8b250c7bc4db9e15f18e11561ef8d
3
+ size 93644160
ITMKNN/metrics.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_val_recall": 0.939646649967552,
3
+ "best_epoch": 19,
4
+ "test": {
5
+ "precision": 0.09556335780402173,
6
+ "recall": 0.9556335780402171,
7
+ "f1": 0.17375155964367578,
8
+ "map": 0.872031991294102,
9
+ "ndcg": 0.8932185284730066
10
+ }
11
+ }
ITMKNN/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
+ itemknn,bpr,exp-itemknn-2,2025-09-03T10:07:41,19,0.939646649967552,0.09556335780402173,0.9556335780402171,0.17375155964367578,0.872031991294102,0.8932185284730066,30,64,0.003,1e-06,1024,10,99,42,200,3,none,0.1,1.0,200,10,1000
ITMKNN/train_log.csv ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ epoch,train_loss,val_precision,val_recall,val_f1,val_map,val_ndcg,is_best
2
+ 1,0.0,0.09392524214727166,0.9392524214727164,0.17077316754049388,0.826599222306074,0.8551118519369406,1
3
+ 2,0.0,0.09377118978159742,0.9377118978159742,0.1704930723301771,0.8239171303785352,0.8527153010214881,0
4
+ 3,0.0,0.09388218026552808,0.9388218026552806,0.17069487321005097,0.8249597588496114,0.8537744231323687,0
5
+ 4,0.0,0.09385064198594124,0.9385064198594121,0.17063753088352948,0.825877387043909,0.854392231151882,0
6
+ 5,0.0,0.09390280144833484,0.9390280144833484,0.17073236626969973,0.8258413312619731,0.8544908137762823,0
7
+ 6,0.0,0.09380758010419761,0.9380758010419762,0.17055923655308658,0.8246766267364704,0.8533826169500904,0
8
+ 7,0.0,0.09391614456662159,0.9391614456662158,0.1707566264847665,0.8265238567927206,0.8550454757000627,0
9
+ 8,0.0,0.09387247617950134,0.9387247617950133,0.17067722941727517,0.8257820934967508,0.854374557453961,0
10
+ 9,0.0,0.09382638177087442,0.938263817708744,0.1705934214015898,0.825216207132946,0.8538319269312203,0
11
+ 10,0.0,0.09386277209347463,0.9386277209347461,0.17065958562449932,0.8253584085761015,0.8540386592574388,0
12
+ 11,0.0,0.09388521279241144,0.9388521279241141,0.17070038689529343,0.826173510887301,0.8546872956368311,0
13
+ 12,0.0,0.09393555273867504,0.9393555273867503,0.17079191407031824,0.825562406178191,0.8543691195968114,1
14
+ 13,0.0,0.09379969553430094,0.9379969553430091,0.17054490097145622,0.8253464926708638,0.8538670956579595,0
15
+ 14,0.0,0.09389067134080147,0.9389067134080143,0.17071031152872987,0.825242031746802,0.8540205855118954,0
16
+ 15,0.0,0.09389249085693145,0.9389249085693144,0.1707136197398753,0.8259863101146707,0.8545761893984116,0
17
+ 16,0.0,0.09390280144833485,0.9390280144833484,0.17073236626969973,0.825117917160415,0.853964160105186,0
18
+ 17,0.0,0.09386034607196794,0.9386034607196793,0.1706551746763053,0.8253522304042684,0.8540177439827326,0
19
+ 18,0.0,0.09384821596443454,0.9384821596443452,0.17063311993533548,0.825884626600151,0.8543947716301715,0
20
+ 19,0.0,0.09396466499675521,0.939646649967552,0.17084484544864578,0.826010591990644,0.854773358701983,1
21
+ 20,0.0,0.09388581929778808,0.9388581929778808,0.17070148963234197,0.8258498608455244,0.8544580577155112,0
22
+ 21,0.0,0.09381789069560104,0.9381789069560101,0.17057798308291094,0.8253482784922507,0.8539138171844203,0
23
+ 22,0.0,0.09393858526555839,0.9393858526555838,0.17079742775556064,0.8249353542285025,0.8539103133790477,0
24
+ 23,0.0,0.09382698827625108,0.9382698827625107,0.17059452413863826,0.8246934909554166,0.85343276427338,0
25
+ 24,0.0,0.09388885182467144,0.9388885182467143,0.1707070033175844,0.826030435787591,0.8546145147222424,0
26
+ 25,0.0,0.09389855591069818,0.9389855591069814,0.17072464711036026,0.8254242673583524,0.8541750277537477,0
27
+ 26,0.0,0.0938773282225147,0.9387732822251469,0.17068605131366305,0.8264168543223184,0.8548587719350986,0
28
+ 27,0.0,0.09388642580316477,0.9388642580316474,0.17070259236939045,0.8259749934945076,0.8545579179740657,0
29
+ 28,0.0,0.0939118990289849,0.9391189902898489,0.17074890732542705,0.8273726057117212,0.85567140737443,0
30
+ 29,0.0,0.09395192838384514,0.9395192838384512,0.17082168797062747,0.8253750369318451,0.8542719116411905,0
31
+ 30,0.0,0.09392948768490834,0.9392948768490833,0.17078088669983332,0.8253999806688443,0.8542313515926303,0
ITMKNN/user_embeddings.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:916bf2a0c2714bfec594aa7063f12d5e9c09eb00d7cb7a8bc6cd7f42cffe22fd
3
+ size 66570112
MF/item_embeddings.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a43d1fa72d6a61d9c231041999f5a06e6bc8b250c7bc4db9e15f18e11561ef8d
3
+ size 93644160
MF/user_embeddings.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:916bf2a0c2714bfec594aa7063f12d5e9c09eb00d7cb7a8bc6cd7f42cffe22fd
3
+ size 66570112
NEUMF/item_embeddings.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a43d1fa72d6a61d9c231041999f5a06e6bc8b250c7bc4db9e15f18e11561ef8d
3
+ size 93644160
NEUMF/user_embeddings.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:916bf2a0c2714bfec594aa7063f12d5e9c09eb00d7cb7a8bc6cd7f42cffe22fd
3
+ size 66570112