moving files

shenyangHuang · Jun 12, 2024 · 47bf81d · 47bf81d
1 parent cb952d7
commit 47bf81d
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 3 deletions.
diff --git a/STHN_README.md → ...linkproppred/thgl-software/STHN_README.md b/STHN_README.md → ...linkproppred/thgl-software/STHN_README.md
diff --git a/stats_figures/create_relation_figures.py b/stats_figures/create_relation_figures.py
@@ -18,7 +18,7 @@
 
 # specify params
 # which datasets
-names = ['thgl-software']#['tkgl-smallpedia','tkgl-polecat', 'tkgl-icews']#[ 'tkgl-polecat', 'tkgl-icews', 'tkgl-wikidata', 'tkgl-smallpedia', 'thgl-software'] #['thgl-software']#
+names = ['tkgl-wikidata']#['tkgl-smallpedia','tkgl-polecat', 'tkgl-icews']#[ 'tkgl-polecat', 'tkgl-icews', 'tkgl-wikidata', 'tkgl-smallpedia', 'thgl-software'] #['thgl-software']#
 # names = [ 'tkgl-polecat', 'tkgl-icews', 'tkgl-wikidata', 'tkgl-smallpedia','tkgl-polecat'] #'tkgl-polecat','tkgl-smallpedia',  'tkgl-yago',  'tkgl-icews' ,'tkgl-smallpedia','thgl-myket','tkgl-yago',  'tkgl-icews','thgl-github', 'thgl-forum', 'tkgl-wikidata']
 # which methods for the mrr_per_rel figures
 methods = ['recurrency', 'regcn', 'cen'] #'recurrency'

diff --git a/tgb/linkproppred/evaluate.py b/tgb/linkproppred/evaluate.py
@@ -108,8 +108,8 @@ def _eval_hits_and_mrr(self, y_pred_pos, y_pred_neg, type_info, k_value):
 
         else:
             y_pred_pos = y_pred_pos.reshape(-1, 1)
-            optimistic_rank = (y_pred_neg >= y_pred_pos).sum(axis=1)
-            pessimistic_rank = (y_pred_neg > y_pred_pos).sum(axis=1)
+            optimistic_rank = (y_pred_neg > y_pred_pos).sum(axis=1)
+            pessimistic_rank = (y_pred_neg >= y_pred_pos).sum(axis=1)
             ranking_list = 0.5 * (optimistic_rank + pessimistic_rank) + 1
             hitsK_list = (ranking_list <= k_value).astype(np.float32)
             mrr_list = 1./ranking_list.astype(np.float32)

diff --git a/sampler_core.cpp → tgb_modules/sampler_core.cpp b/sampler_core.cpp → tgb_modules/sampler_core.cpp
diff --git a/sthn_sampler_setup.py → tgb_modules/sthn_sampler_setup.py b/sthn_sampler_setup.py → tgb_modules/sthn_sampler_setup.py