Np.bincount: valueerror: объект слишком глубок для желаемого массива

Я хочу напечатать количество экземпляров в каждом классе с помощью np.bincount

x_train, x_test, y_train, y_test = train_test_split(data, labels, test_size=0.2)
print(f"Numbers of test instances by class: {np.bincount(y_test)}")
print(f"Numbers of training instances by class: {np.bincount(y_train)}")

но я получаю эту ошибку:

ValueError: object too deep for desired array

Формы x и y:

x_train shape (4415, 32, 32, 1)
x_test shape (1104, 32, 32, 1)
y_train shape (4415, 3)
y_test shape (1104, 3)

Проверьте документы. Я не думаю, что bincount может работать на 2d массивах.

hpaulj 30.03.2021 00:51
0
1
11
0

Другие вопросы по теме