У меня есть файлы .npy, которые содержат 3 dim массива. После применения изменения формы к массивам, он все еще не работает. Вот код.
ta = np.load('class_a.npy', mmap_mode='r')
tb = np.load('class_b.npy', mmap_mode='r')
clf = neighbors.KNeighborsClassifier()
ta.reshape(-1)
tb.reshape(-1)
X_train, X_test, y_train, y_test = train_test_split(ta, tb, test_size=0.2)
clf.fit(X_train, y_train)
Это ошибка, которую я получаю
ValueError: Found array with dim 3. Estimator expected <= 2.
Что я делаю не так?