Reduced number of testing epochs to 50.

ANN_Training.py

@@ -10,7 +10,7 @@ TODO: Fix random seed
 TODO: Write-protect all data and models
 TODO: Put legend outside plot (bbox_to_anchor)
 TODO: Put plotting into separate function
-TODO: Reduce number of testing epochs to 50
+TODO: Reduce number of testing epochs to 50 -> Done
 
 """
 import numpy as np
@@ -205,7 +205,7 @@ class ModelTrainer(object):
             Dictionary containing classification evaluation data.
 
         """
-        self.epoch_training(training_set, num_epochs=100, verbose=False)
+        self.epoch_training(training_set, num_epochs=50, verbose=False)
         self._model.eval()
 
         x_test, y_test = test_set