Я работаю с набором данных PlantVillage для модели классификации изображений, созданной с помощью Keras, на основе этого примера. Я пытаюсь разделить набор данных на поезд и проверить. Тем не менее, у меня есть 1676 изображений и только 15 ярлыков, каждое из которых соответствует названиям папок, содержащих изображения. Вот фрагмент кода, с которым я работаю:
x_train, x_test, y_train, y_test = train_test_split(np_image_list, image_labels, test_size=0.2, random_state=42)
И ошибка, которую я получаю:
ValueError: Found input variables with inconsistent numbers of samples: [1676, 15]