From 6fd7daeef95e3dff552d7a0bb8b6a8fdab5a6a69 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?K=C3=BChle=2C=20Laura=20Christine=20=28lakue103=29?=
 <laura.kuehle@uni-duesseldorf.de>
Date: Tue, 18 Jan 2022 17:57:44 +0100
Subject: [PATCH] Adapted troubled cell detection to new ANN classification.

---
 ANN_Training.py           | 4 +++-
 Troubled_Cell_Detector.py | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/ANN_Training.py b/ANN_Training.py
index 34cca6c..b52964c 100644
--- a/ANN_Training.py
+++ b/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
diff --git a/Troubled_Cell_Detector.py b/Troubled_Cell_Detector.py
index 74b75e1..ef91fcc 100644
--- a/Troubled_Cell_Detector.py
+++ b/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)
-- 
GitLab