From 6c71e398e16feba4d10e3333d91bc1fe33d46013 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: Wed, 24 Nov 2021 14:57:40 +0100
Subject: [PATCH] Changed order of methods.

---
 ANN_Data_Generator.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/ANN_Data_Generator.py b/ANN_Data_Generator.py
index 45f9613..3ecc008 100644
--- a/ANN_Data_Generator.py
+++ b/ANN_Data_Generator.py
@@ -6,6 +6,7 @@ TODO: Improve '_generate_cell_data'
 TODO: Extract normalization (Combine smooth and troubled before normalizing) -> Done
 TODO: Adapt code to generate both normalized and non-normalized data -> Done
 TODO: Improve verbose output
+TODO: Change order of methods -> Done
 
 """
 
@@ -52,11 +53,6 @@ class TrainingDataGenerator(object):
         self._save_data(data_dict)
         return data_dict
 
-    def _save_data(self, data):
-        for key in data.keys():
-            name = self._data_dir + '/' + key + '_data.npy'
-            np.save(name, data[key])
-
     def _calculate_data_set(self, num_samples):
         num_smooth_samples = round(num_samples * self._balance)
         smooth_input, smooth_output = self._generate_cell_data(num_smooth_samples,
@@ -165,6 +161,11 @@ class TrainingDataGenerator(object):
             normalized_input_data[i] = input_data[i] / max_function_value
         return normalized_input_data
 
+    def _save_data(self, data):
+        for key in data.keys():
+            name = self._data_dir + '/' + key + '_data.npy'
+            np.save(name, data[key])
+
 
 # Get Training/Validation Datasets
 np.random.seed(1234)
-- 
GitLab