diff --git a/scripts/tcd/ANN_Data_Generator.py b/scripts/tcd/ANN_Data_Generator.py index 33f7cf3ab1d7d70d2de2d9c8156b2a84f1c8ea57..184a93dad910942100c5607f1c7b26b456d273f9 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):