diff --git a/citeseer/python/citeseer.py b/citeseer/python/citeseer.py
index e8bb8f5f..9f55894b 100755
--- a/citeseer/python/citeseer.py
+++ b/citeseer/python/citeseer.py
@@ -8,7 +8,8 @@
 from pslpython.rule import Rule
 
 MODEL_NAME = 'citeseer'
-DATA_DIR = os.path.join('..', 'data', MODEL_NAME, '0')
+SPLIT = '0'
+DATA_DIR = os.path.join('..', 'data', MODEL_NAME, SPLIT)
 
 ADDITIONAL_PSL_OPTIONS = {
     'log4j.threshold': 'INFO'
diff --git a/cora/python/cora.py b/cora/python/cora.py
index c7071f33..f18bda07 100755
--- a/cora/python/cora.py
+++ b/cora/python/cora.py
@@ -8,7 +8,8 @@
 from pslpython.rule import Rule
 
 MODEL_NAME = 'cora'
-DATA_DIR = os.path.join('..', 'data', MODEL_NAME, '0')
+SPLIT = '0'
+DATA_DIR = os.path.join('..', 'data', MODEL_NAME, SPLIT)
 
 ADDITIONAL_PSL_OPTIONS = {
     'log4j.threshold': 'INFO'
diff --git a/entity-resolution/python/entity-resolution.py b/entity-resolution/python/entity-resolution.py
index 22e05051..772b4ec8 100755
--- a/entity-resolution/python/entity-resolution.py
+++ b/entity-resolution/python/entity-resolution.py
@@ -8,7 +8,8 @@
 from pslpython.rule import Rule
 
 MODEL_NAME = 'entity-resolution'
-DATA_DIR = os.path.join('..', 'data', MODEL_NAME)
+SPLIT = '0'
+DATA_DIR = os.path.join('..', 'data', MODEL_NAME, SPLIT)
 
 ADDITIONAL_PSL_OPTIONS = {
     'log4j.threshold': 'INFO'
diff --git a/friendship/python/friendship.py b/friendship/python/friendship.py
index 9cd138bc..3ef1da62 100755
--- a/friendship/python/friendship.py
+++ b/friendship/python/friendship.py
@@ -8,7 +8,8 @@
 from pslpython.rule import Rule
 
 MODEL_NAME = 'friendship'
-DATA_DIR = os.path.join('..', 'data', MODEL_NAME)
+SPLIT = '0'
+DATA_DIR = os.path.join('..', 'data', MODEL_NAME, SPLIT, 'eval')
 
 ADDITIONAL_PSL_OPTIONS = {
     'log4j.threshold': 'INFO'
diff --git a/knowledge-graph-identification/python/knowledge-graph-identification.py b/knowledge-graph-identification/python/knowledge-graph-identification.py
index 79450d9a..332dc198 100755
--- a/knowledge-graph-identification/python/knowledge-graph-identification.py
+++ b/knowledge-graph-identification/python/knowledge-graph-identification.py
@@ -9,7 +9,9 @@
 from pslpython.rule import Rule
 
 MODEL_NAME = 'knowledge-graph-identification'
-DATA_DIR = os.path.join('..', 'data', 'kgi')
+MODEL_SHORT_NAME = 'kgi'
+SPLIT = '0'
+DATA_DIR = os.path.join('..', 'data', MODEL_SHORT_NAME, SPLIT)
 
 ADDITIONAL_PSL_OPTIONS = {
     'log4j.threshold': 'INFO'
diff --git a/simple-acquaintances/python/simple-acquaintances.py b/simple-acquaintances/python/simple-acquaintances.py
index 5d5feff1..1b065fb9 100755
--- a/simple-acquaintances/python/simple-acquaintances.py
+++ b/simple-acquaintances/python/simple-acquaintances.py
@@ -8,7 +8,8 @@
 from pslpython.rule import Rule
 
 MODEL_NAME = 'simple-acquaintances'
-DATA_DIR = os.path.join('..', 'data')
+SPLIT = '0'
+DATA_DIR = os.path.join('..', 'data', MODEL_NAME, SPLIT, 'eval')
 
 ADDITIONAL_PSL_OPTIONS = {
     'log4j.threshold': 'INFO'
diff --git a/social-network-analysis/python/social-network-analysis.py b/social-network-analysis/python/social-network-analysis.py
index 3dcaa727..de7865cb 100755
--- a/social-network-analysis/python/social-network-analysis.py
+++ b/social-network-analysis/python/social-network-analysis.py
@@ -8,7 +8,8 @@
 from pslpython.rule import Rule
 
 MODEL_NAME = 'social-network-analysis'
-DATA_DIR = os.path.join('..', 'data', MODEL_NAME)
+SPLIT = '0'
+DATA_DIR = os.path.join('..', 'data', MODEL_NAME, SPLIT, 'eval')
 
 ADDITIONAL_PSL_OPTIONS = {
     'log4j.threshold': 'INFO'
diff --git a/trust-prediction/python/trust-prediction.py b/trust-prediction/python/trust-prediction.py
index 2af1c93e..2819a621 100755
--- a/trust-prediction/python/trust-prediction.py
+++ b/trust-prediction/python/trust-prediction.py
@@ -8,7 +8,8 @@
 from pslpython.rule import Rule
 
 MODEL_NAME = 'trust-prediction'
-DATA_DIR = os.path.join('..', 'data', MODEL_NAME)
+SPLIT = '0'
+DATA_DIR = os.path.join('..', 'data', MODEL_NAME, SPLIT)
 
 ADDITIONAL_PSL_OPTIONS = {
     'log4j.threshold': 'INFO'
diff --git a/user-modeling/python/user-modeling.py b/user-modeling/python/user-modeling.py
index afe11322..99428397 100755
--- a/user-modeling/python/user-modeling.py
+++ b/user-modeling/python/user-modeling.py
@@ -8,7 +8,8 @@
 from pslpython.rule import Rule
 
 MODEL_NAME = 'user-modeling'
-DATA_DIR = os.path.join('..', 'data', MODEL_NAME)
+SPLIT = '0'
+DATA_DIR = os.path.join('..', 'data', MODEL_NAME, SPLIT, 'eval')
 
 ADDITIONAL_PSL_OPTIONS = {
     'log4j.threshold': 'INFO'