Upload folder using huggingface_hub
Browse files- ITMKNN/REPORT.txt +5 -0
- ITMKNN/hparams.json +28 -0
- ITMKNN/item_embeddings.npy +3 -0
- ITMKNN/metrics.json +11 -0
- ITMKNN/test_metrics.csv +2 -0
- ITMKNN/train_log.csv +31 -0
- ITMKNN/user_embeddings.npy +3 -0
- MF/item_embeddings.npy +3 -0
- MF/user_embeddings.npy +3 -0
- NEUMF/item_embeddings.npy +3 -0
- NEUMF/user_embeddings.npy +3 -0
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
|