<reponame>oercompbiomed/CBM101
print(np.bincount(y_train))
print(np.bincount(y_test))

# yes, they are more or less equal
print(np.bincount(y_train))
print(np.bincount(y_test))

# yes, they are more or less equal
# yes, they are more or less equal