Add to run on command line

Not just Intellij now
jacobbieker · Feb 10, 2019 · 48e3574 · 48e3574
1 parent 9cfb670
commit 48e3574
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 0 deletions.
diff --git a/examples/talos_3d_classification.py b/examples/talos_3d_classification.py
@@ -9,6 +9,9 @@
 from talos.model.normalizers import lr_normalizer
 from talos.metrics.keras_metrics import root_mean_squared_error, fmeasure_acc, matthews_correlation_acc, precision_acc, recall_acc
 
+from os import sys, path
+sys.path.append(path.dirname(path.dirname(path.abspath(__file__))))
+
 from factnn.utils.cross_validate import get_chunk_of_data
 
 # Parameter Dictionary for talos

diff --git a/examples/talos_flat_classification.py b/examples/talos_flat_classification.py
@@ -10,6 +10,9 @@
 from talos.metrics.keras_metrics import root_mean_squared_error, fmeasure_acc, matthews_correlation_acc, precision_acc, \
     recall_acc
 
+from os import sys, path
+sys.path.append(path.dirname(path.dirname(path.abspath(__file__))))
+
 from factnn.utils.cross_validate import get_chunk_of_data, get_data_generators
 
 # Get arguments

diff --git a/examples/talos_time_classification.py b/examples/talos_time_classification.py
@@ -9,6 +9,8 @@
 from talos.model.normalizers import lr_normalizer
 from talos.metrics.keras_metrics import root_mean_squared_error, fmeasure_acc, matthews_correlation_acc, precision_acc, recall_acc
 
+from os import sys, path
+sys.path.append(path.dirname(path.dirname(path.abspath(__file__))))
 
 from factnn.utils.cross_validate import get_chunk_of_data, get_data_generators
 

diff --git a/examples/test_event_files.py b/examples/test_event_files.py
@@ -1,4 +1,6 @@
 import os
+from os import sys, path
+sys.path.append(path.dirname(path.dirname(path.abspath(__file__))))
 os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"   # see issue #152
 os.environ["CUDA_VISIBLE_DEVICES"] = ""
 from factnn import ProtonPreprocessor, GammaPreprocessor, GammaDiffusePreprocessor