 `y_train[y_train[:] == 1]` and `y_train[y_train[:] == 0]` both return 537 rows.