Adapted troubled cell detection to new ANN classification.

ANN_Training.py

@@ -2,9 +2,11 @@
 """
 @author: Laura C. Kühle, Soraya Terrab (sorayaterrab)
 
+TODO: Test new ANN set-up with Soraya
+TODO: Integrate Main.py into Snakefile
 TODO: Add log to pipeline -> Done
 TODO: Remove object set-up (for more flexibility)
-TODO: Adapt TCD file to new classification
+TODO: Adapt TCD file to new classification -> Done
 TODO: Add documentation
 TODO: Improve log output
 TODO: Remove unnecessary comments -> Done

Troubled_Cell_Detector.py

@@ -265,7 +265,7 @@ class ArtificialNeuralNetwork(TroubledCellDetector):
             'input_size': self._stencil_len+2, 'first_hidden_size': 8, 'second_hidden_size': 4,
             'output_size': 2, 'activation_function': 'Softmax', 'activation_config': {'dim': 1}})
         self._model_state = config.pop(
-            'model_state', 'Train24k24k_Valid8k8k_Norm12ReLU8+4nodesSM1Adamlr1e-2MSE.pt')
+            'model_state', 'Snakemake-Test/trained models/model__Adam.pt')
 
         if not hasattr(ANN_Model, self._model):
             raise ValueError('Invalid model: "%s"' % self._model)