update description

shenyangHuang · May 31, 2024 · 4846e06 · 4846e06
1 parent c96805f
commit 4846e06
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 5 deletions.
diff --git a/stats_figures/README.md b/stats_figures/README.md
@@ -1,9 +1,13 @@
 ### How to compute stats and figures
-for each of them, you can specify the datasets of interest in the list at the beginning.
-same applies to methods of interest, if it is about correlation of methods results and stats.
+
+You run each of the scripts running 'python stats_figures/nameofthescript.py'
+Each of them computes the stats or creates the figures automatically for all datasets. If you want to only focus on some subset of datasets you need to specify that in the list
+'names = [ 'tkgl-polecat', ... ]' at the beginning of each script.
+The same applies to methods of interest, if it is about correlation of methods results and stats.
 
 For *creating the figures* you only need to run 2. and 5., provided that you have previously stored all stats in the respective dataset subfolder.
 
+
 ## 1. compute_dataset_stats.py
 - loads datasets and computes all stats that we put in paper table and writes it in stats_figures/dataset_name/dataset_stats.csv 
 - computes number of nodes and stores in stats_figures/dataset_name/figs/numedges_datasetname.json # number of edges per timestep (to create the figures)

diff --git a/stats_figures/compute_dataset_stats.py b/stats_figures/compute_dataset_stats.py
@@ -28,7 +28,7 @@
 
 
 # specify all datasets
-names = ['thgl-myket'] #, 'tkgl-smallpedia','tkgl-polecat',  'thgl-software', 'tkgl-icews','thgl-github', 'thgl-forum', 'tkgl-wikidata', 'thgl-myket','tkgl-yago']
+names = ['thgl-myket', 'tkgl-smallpedia','tkgl-polecat',  'thgl-software', 'tkgl-icews','thgl-github', 'thgl-forum', 'tkgl-wikidata', 'thgl-myket','tkgl-yago']
 for dataset_name in names:
     current_dir = os.path.dirname(os.path.abspath(__file__))
     # Navigate one folder up

diff --git a/stats_figures/compute_relation_dataset_stats.py b/stats_figures/compute_relation_dataset_stats.py
@@ -17,7 +17,7 @@
 import stats_figures.dataset_utils as du
 
 # specify params
-names = [ 'tkgl-polecat'] #'tkgl-polecat','tkgl-smallpedia',  'tkgl-yago',  'tkgl-icews' ,'tkgl-smallpedia','thgl-myket','tkgl-yago',  'tkgl-icews','thgl-github', 'thgl-forum', 'tkgl-wikidata']
+names = [ 'tkgl-polecat','tkgl-polecat','tkgl-smallpedia',  'tkgl-yago',  'tkgl-icews' ,'tkgl-smallpedia','thgl-myket','tkgl-yago',  'tkgl-icews','thgl-github', 'thgl-forum', 'tkgl-wikidata']
 colortgb = '#60ab84' #tgb logo colrs
 colortgb2 = '#eeb641'
 colortgb3 = '#dd613a'

diff --git a/stats_figures/compute_relation_results_df.py b/stats_figures/compute_relation_results_df.py
@@ -19,7 +19,7 @@
 
 
 # specify params
-names = [ 'tkgl-polecat'] #'tkgl-icews', 'tkgl-polecat', 'tkgl-smallpedia'] #'tkgl-polecat','tkgl-smallpedia',  'tkgl-yago',  'tkgl-icews' ,'tkgl-smallpedia','thgl-myket','tkgl-yago',  'tkgl-icews','thgl-github', 'thgl-forum', 'tkgl-wikidata']
+names = [ 'tkgl-polecat', 'tkgl-icews', 'tkgl-polecat', 'tkgl-smallpedia'] #'tkgl-polecat','tkgl-smallpedia',  'tkgl-yago',  'tkgl-icews' ,'tkgl-smallpedia','thgl-myket','tkgl-yago',  'tkgl-icews','thgl-github', 'thgl-forum', 'tkgl-wikidata']
 methods = ['recurrency', 'regcn', 'cen'] #'recurrency'
 
 # this is where the results per relation are stored