Skip to content

Commit 6b95961

Browse files
committed
2 parents 0075fe1 + 8b33b24 commit 6b95961

File tree

5 files changed

+110
-1
lines changed

5 files changed

+110
-1
lines changed

charts/MAP@5_new.pdf

17.2 KB
Binary file not shown.

charts/nDCG@5_new.pdf

17.6 KB
Binary file not shown.

docs/source/benchmark_results/results.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ Experiment ran using the MovieLens-100k dataset with the following presented mod
2222
|RotatE based model + cosine similarity;embedding_dim=200;epochs=25;seed=42;triples=all|.0061 ± .0007|.0093 ± .0013|
2323
|EPHEN based model + cosine similarity;embedding_model=sentence-transformers/all-roberta-large-v1;embed_with=abstract;iterations=30;mi=0.5|.0108 ± .0011|.0155 ± .0013|
2424
|EPHEN based model + cosine similarity;embedding_model=deepwalk_based;embedding_model_kwargs={'walk_len': 10, 'p': 1.0, 'q': 1.0, 'n_walks': 50, 'embedding_size': 64, 'epochs': 1};embed_with=graph;iterations=30;mi=0.5|.1203 ± .0065|.1648 ± .0072|
25+
|Entity2Rec;embedding_model=deepwalk_based;embedding_model_kwargs={'config': {'save_weights': True}, 'parameters': {'walk_len': 10, 'p': 1.0, 'q': 1.0, 'n_walks': 50, 'embedding_size': 64, 'epochs': 1}};run_all=False;workers=6;iterations=1;collab_only=False;content_only=False|.0082 ± .0017|.0129 ± .0026|
2526

2627
- Summarized execution time results from `experiment_results/ml-100k_times.csv` (configuration: CPU: AMD EPYC 7502P 32-Core Processor; RAM: 94GB; GPUs: ['NVIDIA A2']):
2728

@@ -40,6 +41,7 @@ Experiment ran using the MovieLens-100k dataset with the following presented mod
4041
|RotatE based model + cosine similarity;embedding_dim=200;epochs=25;seed=42;triples=all|48.31 ± 1.420|
4142
|EPHEN based model + cosine similarity;embedding_model=sentence-transformers/all-roberta-large-v1;embed_with=abstract;iterations=30;mi=0.5|85.05 ± .3143|
4243
|EPHEN based model + cosine similarity;embedding_model=deepwalk_based;embedding_model_kwargs={'walk_len': 10, 'p': 1.0, 'q': 1.0, 'n_walks': 50, 'embedding_size': 64, 'epochs': 1};embed_with=graph;iterations=30;mi=0.5|50.65 ± .3341|
44+
|Entity2Rec;embedding_model=deepwalk_based;embedding_model_kwargs={'config': {'save_weights': True}, 'parameters': {'walk_len': 10, 'p': 1.0, 'q': 1.0, 'n_walks': 50, 'embedding_size': 64, 'epochs': 1}};run_all=False;workers=6;iterations=1;collab_only=False;content_only=False|73506 ± 5979.|
4345

4446
## ml-100k_enriched
4547

@@ -62,6 +64,7 @@ Experiment ran using the MovieLens-100k dataset with DBpedia enrichement and the
6264
|RotatE based model + cosine similarity;embedding_dim=200;epochs=25;seed=42;triples=all|.0065 ± .0006|.0104 ± .0010|
6365
|EPHEN based model + cosine similarity;embedding_model=sentence-transformers/all-roberta-large-v1;embed_with=abstract;iterations=30;mi=0.5|.0141 ± .0006|.0252 ± .0005|
6466
|EPHEN based model + cosine similarity;embedding_model=deepwalk_based;embedding_model_kwargs={'walk_len': 10, 'p': 1.0, 'q': 1.0, 'n_walks': 50, 'embedding_size': 64, 'epochs': 1};embed_with=graph;iterations=30;mi=0.5|.1812 ± .0078|.2338 ± .0076|
67+
|Entity2Rec;embedding_model=deepwalk_based;embedding_model_kwargs={'config': {'save_weights': True}, 'parameters': {'walk_len': 10, 'p': 1.0, 'q': 1.0, 'n_walks': 50, 'embedding_size': 64, 'epochs': 1}};run_all=False;workers=6;iterations=1;collab_only=False;content_only=False|.0076 ± .0010|.0120 ± .0019|
6568

6669
- Summarized execution time results from `experiment_results/ml-100k_enriched_times.csv` (configuration: CPU: AMD EPYC 7502P 32-Core Processor; RAM: 94GB; GPUs: ['NVIDIA A2']):
6770

@@ -80,6 +83,7 @@ Experiment ran using the MovieLens-100k dataset with DBpedia enrichement and the
8083
|RotatE based model + cosine similarity;embedding_dim=200;epochs=25;seed=42;triples=all|102.9 ± 1.780|
8184
|EPHEN based model + cosine similarity;embedding_model=sentence-transformers/all-roberta-large-v1;embed_with=abstract;iterations=30;mi=0.5|177.4 ± 1.292|
8285
|EPHEN based model + cosine similarity;embedding_model=deepwalk_based;embedding_model_kwargs={'walk_len': 10, 'p': 1.0, 'q': 1.0, 'n_walks': 50, 'embedding_size': 64, 'epochs': 1};embed_with=graph;iterations=30;mi=0.5|96.80 ± 1.006|
86+
|Entity2Rec;embedding_model=deepwalk_based;embedding_model_kwargs={'config': {'save_weights': True}, 'parameters': {'walk_len': 10, 'p': 1.0, 'q': 1.0, 'n_walks': 50, 'embedding_size': 64, 'epochs': 1}};run_all=False;workers=6;iterations=1;collab_only=False;content_only=False|68685 ± 10848|
8387

8488
## ml-1m
8589

docs/source/imgs/framework.svg

Lines changed: 1 addition & 1 deletion
Loading

test.csv

Lines changed: 105 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,105 @@
1+
classifier_name,dataset_name,accuracy
2+
Node2Vec_0,ml-100k.csv,0.1641552384592829
3+
Node2Vec_1,ml-100k.csv,0.1621669265350805
4+
TransE_2,ml-100k.csv,0.0068440086811613
5+
TransH_3,ml-100k.csv,0.0016045649166517
6+
TransR_4,ml-100k.csv,0.006316580257412
7+
TransD_5,ml-100k.csv,0.0096928490773496
8+
TuckER_6,ml-100k.csv,0.0065033209967898
9+
RESCAL_7,ml-100k.csv,0.0128138879039556
10+
DistMult_8,ml-100k.csv,0.0105667805930337
11+
ComplEx_9,ml-100k.csv,0.0105675659209028
12+
RotatE_10,ml-100k.csv,0.0093335263854372
13+
EPHEN_11,ml-100k.csv,0.0155369900340128
14+
EPHEN_12,ml-100k.csv,0.1647689211271926
15+
Node2Vec_0,ml-1m.csv,0.1451855552197163
16+
Node2Vec_1,ml-1m.csv,0.1443807826814608
17+
TransE_2,ml-1m.csv,0.0059231105779098
18+
TransH_3,ml-1m.csv,0.0022034352558949
19+
TransR_4,ml-1m.csv,0.0016216214820227
20+
TransD_5,ml-1m.csv,0.0033781562985516
21+
TuckER_6,ml-1m.csv,0.000270349638519
22+
RESCAL_7,ml-1m.csv,0.0115909215084142
23+
DistMult_8,ml-1m.csv,0.0018514195522373
24+
ComplEx_9,ml-1m.csv,0.0023048662594048
25+
RotatE_10,ml-1m.csv,0.005665889603762
26+
EPHEN_11,ml-1m.csv,0.0073489827197795
27+
EPHEN_12,ml-1m.csv,0.1437505674384252
28+
Node2Vec_0,lastfm.csv,0.1628311998902248
29+
Node2Vec_1,lastfm.csv,0.1620902921598351
30+
TransE_2,lastfm.csv,0.0003161278486355
31+
TransH_3,lastfm.csv,0.0002443629978509
32+
TransR_4,lastfm.csv,0.0003454078819604
33+
TransD_5,lastfm.csv,0.0003049977656416
34+
TuckER_6,lastfm.csv,0.0003542394025717
35+
RESCAL_7,lastfm.csv,0.0003751940708921
36+
DistMult_8,lastfm.csv,0.0003055808117677
37+
ComplEx_9,lastfm.csv,0.0003764969292397
38+
RotatE_10,lastfm.csv,0.0003465439720528
39+
EPHEN_11,lastfm.csv,9.766239761934156e-05
40+
EPHEN_12,lastfm.csv,0.1624390660293245
41+
Node2Vec_0,douban-movie.csv,0.6753598020415168
42+
Node2Vec_1,douban-movie.csv,0.682548986273064
43+
TransE_2,douban-movie.csv,0.4812658913028417
44+
TransH_3,douban-movie.csv,0.4686972848098014
45+
TransR_4,douban-movie.csv,0.4854895912466208
46+
TransD_5,douban-movie.csv,0.4888088848800759
47+
TuckER_6,douban-movie.csv,0.4903067645862988
48+
RESCAL_7,douban-movie.csv,0.4910420646376879
49+
DistMult_8,douban-movie.csv,0.4877454118453105
50+
ComplEx_9,douban-movie.csv,0.4857444880561369
51+
RotatE_10,douban-movie.csv,0.4881900458711927
52+
EPHEN_11,douban-movie.csv,0.5283584029621593
53+
EPHEN_12,douban-movie.csv,0.6754623832255416
54+
Node2Vec_0,ml-100k_enriched.csv,0.23384413325499
55+
Node2Vec_1,ml-100k_enriched.csv,0.2301353114076907
56+
TransE_2,ml-100k_enriched.csv,0.0067798872336795
57+
TransH_3,ml-100k_enriched.csv,0.001515248085174
58+
TransR_4,ml-100k_enriched.csv,0.0055869572383092
59+
TransD_5,ml-100k_enriched.csv,0.0100525858061356
60+
TuckER_6,ml-100k_enriched.csv,0.0065211670089126
61+
RESCAL_7,ml-100k_enriched.csv,0.0123767420002055
62+
DistMult_8,ml-100k_enriched.csv,0.0060451642279894
63+
ComplEx_9,ml-100k_enriched.csv,0.0105705444304996
64+
RotatE_10,ml-100k_enriched.csv,0.0103525604106293
65+
EPHEN_11,ml-100k_enriched.csv,0.0251651821014341
66+
EPHEN_12,ml-100k_enriched.csv,0.2337932706297812
67+
Node2Vec_0,ml-1m_enriched.csv,0.201402971744723
68+
Node2Vec_1,ml-1m_enriched.csv,0.2016113048091056
69+
TransE_2,ml-1m_enriched.csv,0.0062580385792671
70+
TransH_3,ml-1m_enriched.csv,0.0021040943911566
71+
TransR_4,ml-1m_enriched.csv,0.0012022241380167
72+
TransD_5,ml-1m_enriched.csv,0.0033176395321861
73+
TuckER_6,ml-1m_enriched.csv,0.0003231785741516
74+
RESCAL_7,ml-1m_enriched.csv,0.0116248350256572
75+
DistMult_8,ml-1m_enriched.csv,0.0028605104775144
76+
ComplEx_9,ml-1m_enriched.csv,0.0021708541805602
77+
RotatE_10,ml-1m_enriched.csv,0.0054761209398883
78+
EPHEN_11,ml-1m_enriched.csv,0.0030490892154949
79+
EPHEN_12,ml-1m_enriched.csv,0.2013182942427749
80+
Node2Vec_0,lastfm_enriched.csv,0.1922575542411986
81+
Node2Vec_1,lastfm_enriched.csv,0.1946359525401797
82+
TransE_2,lastfm_enriched.csv,0.0003306550658207
83+
TransH_3,lastfm_enriched.csv,0.0002073160012181
84+
TransR_4,lastfm_enriched.csv,4.449703407980332e-05
85+
TransD_5,lastfm_enriched.csv,0.0003026114647691
86+
TuckER_6,lastfm_enriched.csv,0.0002657454356579
87+
RESCAL_7,lastfm_enriched.csv,0.0002441161478887
88+
DistMult_8,lastfm_enriched.csv,0.000326128569527
89+
ComplEx_9,lastfm_enriched.csv,0.0003718936471592
90+
RotatE_10,lastfm_enriched.csv,0.0002706814944477
91+
EPHEN_11,lastfm_enriched.csv,0.0023167244729388
92+
EPHEN_12,lastfm_enriched.csv,0.1903385684288752
93+
Node2Vec_0,douban-movie_enriched.csv,0.6626822903937484
94+
Node2Vec_1,douban-movie_enriched.csv,0.6609257528235571
95+
TransE_2,douban-movie_enriched.csv,0.4830490089209694
96+
TransH_3,douban-movie_enriched.csv,0.4695172763620502
97+
TransR_4,douban-movie_enriched.csv,0.4871930197925317
98+
TransD_5,douban-movie_enriched.csv,0.4883454258813243
99+
TuckER_6,douban-movie_enriched.csv,0.4865982329777741
100+
RESCAL_7,douban-movie_enriched.csv,0.4915845109665535
101+
DistMult_8,douban-movie_enriched.csv,0.4809811491011413
102+
ComplEx_9,douban-movie_enriched.csv,0.4870980579294152
103+
RotatE_10,douban-movie_enriched.csv,0.4894867098468399
104+
EPHEN_11,douban-movie_enriched.csv,0.4601293122576537
105+
EPHEN_12,douban-movie_enriched.csv,0.6616522518315255

0 commit comments

Comments
 (0)