From a11fbe9419bbf7168fa67cf71aed8584ccaf179a 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: Sat, 8 Oct 2022 23:44:42 +0200 Subject: [PATCH] Vectorized '_normalize_data()' during data generation. --- scripts/tcd/ANN_Data_Generator.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/scripts/tcd/ANN_Data_Generator.py b/scripts/tcd/ANN_Data_Generator.py index 33f7cf3..184a93d 100644 --- a/scripts/tcd/ANN_Data_Generator.py +++ b/scripts/tcd/ANN_Data_Generator.py @@ -260,11 +260,9 @@ class TrainingDataGenerator: Array containing normalized input data. """ - normalized_input_data = [] - for entry in input_data: - max_function_value = max(max(np.absolute(entry)), 1) - normalized_input_data.append(entry / max_function_value) - return np.array(normalized_input_data) + normalized_input_data = input_data/np.maximum( + 1, np.max(np.abs(input_data), axis=1))[:, np.newaxis] + return normalized_input_data @staticmethod def _save_data(directory, data): -- GitLab