Argument loading in training scripts

Experiments/train_eiv.py

@@ -4,6 +4,7 @@ Train EiV model using different seeds
 import random
 import importlib
 import os
+import argparse
 import json
 
 import numpy as np
@@ -15,7 +16,14 @@ from torch.utils.tensorboard.writer import SummaryWriter
 from EIVArchitectures import Networks, initialize_weights
 from EIVTrainingRoutines import train_and_store, loss_functions
 
-data = 'california'
+
+# read in data via --data option
+parser = argparse.ArgumentParser()
+parser.add_argument("--data", help="Loads data", default='california')
+parser.add_argument("--no-autoindent", help="",
+        action="store_true") # to avoid conflics in IPython
+args = parser.parse_args()
+data = args.data
 
 # load hyperparameters from JSON file
 with open(os.path.join('configurations',f'eiv_{data}.json'),'r') as conf_file:
@@ -42,6 +50,8 @@ gamma = conf_dict["gamma"]
 hidden_layers = conf_dict["hidden_layers"]
 seed_range = conf_dict['seed_range']
 
+print(f"Training on {long_dataname} data")
+
 try:
     gpu_number = conf_dict["gpu_number"]
     device = torch.device(f'cuda:{gpu_number}' if torch.cuda.is_available() else 'cpu')

Experiments/train_noneiv.py

@@ -4,6 +4,7 @@ Train non-EiV model using different seeds
 import random
 import importlib
 import os
+import argparse
 import json
 
 import numpy as np
@@ -15,7 +16,14 @@ from torch.utils.tensorboard.writer import SummaryWriter
 from EIVArchitectures import Networks, initialize_weights
 from EIVTrainingRoutines import train_and_store, loss_functions
 
-data = 'california'
+
+# read in data via --data option
+parser = argparse.ArgumentParser()
+parser.add_argument("--data", help="Loads data", default='california')
+parser.add_argument("--no-autoindent", help="",
+        action="store_true") # to avoid conflics in IPython
+args = parser.parse_args()
+data = args.data
 
 # load hyperparameters from JSON file
 with open(os.path.join('configurations',f'noneiv_{data}.json'),'r') as conf_file: