Re-added the Config.process parameters.

4AI · Nov 19, 2017 · bb6b44c · bb6b44c
1 parent 1d99b28
commit bb6b44c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/main.py b/main.py
@@ -42,7 +42,6 @@
 
 #model_name = 'DistMult_{0}_{1}'.format(Config.input_dropout, Config.dropout)
 model_name = '{2}_{0}_{1}'.format(Config.input_dropout, Config.dropout, Config.model_name)
-do_process = True
 epochs = 1000
 load = False
 if Config.dataset is None:
@@ -91,7 +90,7 @@ def preprocess(dataset_name, delete_data=False):
 
 
 def main():
-    if do_process: preprocess(Config.dataset, delete_data=True)
+    if Config.process: preprocess(Config.dataset, delete_data=True)
     input_keys = ['e1', 'rel', 'e2', 'e2_multi1', 'e2_multi2']
     p = Pipeline(Config.dataset, keys=input_keys)
     p.load_vocabs()