Skip to content
Snippets Groups Projects
Commit 3e6ab5e8 authored by Laura Christine Kühle's avatar Laura Christine Kühle
Browse files

Fixed bug for ANN model initialization.

parent 39666197
No related branches found
No related tags found
No related merge requests found
...@@ -3,6 +3,7 @@ ...@@ -3,6 +3,7 @@
@author: Laura C. Kühle, Soraya Terrab (sorayaterrab) @author: Laura C. Kühle, Soraya Terrab (sorayaterrab)
TODO: Fix cell averages and reconstructions to create data with an x-point stencil TODO: Fix cell averages and reconstructions to create data with an x-point stencil
TODO: Fix bug for ANN model initialization -> Done (later)
""" """
import os import os
...@@ -207,7 +208,7 @@ class ArtificialNeuralNetwork(TroubledCellDetector): ...@@ -207,7 +208,7 @@ class ArtificialNeuralNetwork(TroubledCellDetector):
if not hasattr(ANN_Model, self._model): if not hasattr(ANN_Model, self._model):
raise ValueError('Invalid model: "%s"' % self._model) raise ValueError('Invalid model: "%s"' % self._model)
self._model = getattr(ANN_Model, self._model)(**self._model_config) self._model = getattr(ANN_Model, self._model)(self._model_config)
def get_cells(self, projection): def get_cells(self, projection):
num_ghost_cells = self._stencil_len//2 num_ghost_cells = self._stencil_len//2
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment