Added option to change 'num_iterations' for model testing.

config.yaml

@@ -67,6 +67,7 @@ ANN_Data:
 # Parameter for Model Training and Evaluation
 ANN_Training:
   compare_normalization: True
+  num_iterations: 2
 
   classification_colors:
     Accuracy: '#FF00FF'  # magenta

workflows/ANN_training.smk

@@ -24,6 +24,7 @@ rule test_model:
     output:
         DIR+'/model evaluation/classification_accuracy/' + '_'.join(MODELS.keys()) + '.pdf'
     params:
+        num_iterations = config['num_iterations'],
         colors = config['classification_colors'],
         compare_normalization = config['compare_normalization']
     log:
@@ -31,14 +32,14 @@ rule test_model:
     run:
         models = {}
         with open(str(log), 'w') as logfile:
-            sys.stdout = logfile
-            sys.stderr = logfile
+            # sys.stdout = logfile
+            # sys.stderr = logfile
             for model in MODELS:
                 trainer= ANN_Training.ModelTrainer({'model_name': model, 'dir': DIR,
                                                 'model_dir': DIR, **MODELS[model]})
                 models[model] = trainer
-            evaluate_models(models=models, directory=DIR, num_iterations=2, colors=params.colors,
-                compare_normalization=params.compare_normalization)
+            evaluate_models(models=models, directory=DIR, num_iterations=params.num_iterations,
+                colors=params.colors, compare_normalization=params.compare_normalization)
 
 rule train_model:
     input: