Update README.md

Signed-off-by: David Rotermund <54365609+davrot@users.noreply.github.com>
This commit is contained in:
David Rotermund 2024-02-06 18:50:19 +01:00 committed by GitHub
parent d100e52076
commit 9922bc8f68
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -95,7 +95,7 @@ data_data = data_data[idx]
data_class = data_class[idx]
data_cumsum = np.cumsum(data_class)
border = np.argmax(data_cumsum)
border = np.argmax(np.abs(data_cumsum))
variant_a = (data_class[:border] == -1).sum() + (data_class[border:] == +1).sum()