From daadaeb2c65a0216f09ca59eea68b10c8b1a4009 Mon Sep 17 00:00:00 2001 From: Kristofers Solo Date: Fri, 5 Jan 2024 16:22:43 +0200 Subject: [PATCH] feat(ai): add fitness calculation function templates --- src/ai/fitness.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/src/ai/fitness.py b/src/ai/fitness.py index 3cab0d4..2c2e81a 100644 --- a/src/ai/fitness.py +++ b/src/ai/fitness.py @@ -15,6 +15,30 @@ def calculate_fitness(game: Game) -> float: return fitness +def get_peaks(field: np.ndarray) -> int: + pass + + +def get_row_transitions(field: np.ndarray) -> int: + pass + + +def get_col_transitions(field: np.ndarray) -> int: + pass + + +def get_bumpiness(field: np.ndarray) -> int: + pass + + +def get_holes(field: np.ndarray) -> int: + pass + + +def get_wells(field: np.ndarray) -> int: + pass + + def _calc_holes(field: np.ndarray) -> float: height, width = field.shape penalty = 0