From bbdb824e52d2b621334f5691712429d7026674e0 Mon Sep 17 00:00:00 2001 From: Kristofers Solo Date: Fri, 5 Jan 2024 17:13:23 +0200 Subject: [PATCH] feat(ai): add `get_peaks` --- src/ai/fitness/peaks.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 src/ai/fitness/peaks.py diff --git a/src/ai/fitness/peaks.py b/src/ai/fitness/peaks.py new file mode 100644 index 0000000..f67193e --- /dev/null +++ b/src/ai/fitness/peaks.py @@ -0,0 +1,14 @@ +import numpy as np + +from ai.log import log + + +def get_peaks(field: np.ndarray) -> float: + col_num = field.shape[1] + peaks = np.zeros(col_num) + + for col in range(col_num): + if 1 in field[:, col]: + peaks[col] = field.shape[0] - np.argmax(field[:, col], axis=0) + + return float(np.sum(peaks))