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