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