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