kristoferssolo/Tetris
1
0
Fork 0
mirror of https://github.com/kristoferssolo/Tetris.git synced 2025-10-21 20:00:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(ai): add bumpiness

Browse Source
This commit is contained in:
Kristofers Solo 2024-01-09 20:27:56 +02:00
parent 76e306bd8a
commit 89884d7e2a
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/ai/heuristics/bumpiness.py
View File

@ -4,4 +4,13 @@ import numpy as np
def bumpiness(
field: np.ndarray[int, np.dtype[np.uint8]],
) -> int:
pass
"""
Calculate the bumpiness of a given signal based on peaks.
Args:
field: The game field.
Returns:
The bumpiness of the field.
"""
return int(np.sum(np.abs(np.diff(field.shape[0] - np.argmax(field, axis=0)))))