Как решить несогласованное количество выборок при разделении данных вручную в папку?

#scikit-learn #deep-learning #split #confusion-matrix

Вопрос:

Я собираюсь выполнить некоторую работу,такую как матрица путаницы, точность, отзыв и оценка f1. Но я получил ошибку, в которой говорилось

 Found input variables with inconsistent numbers of samples: [56, 1781]  

В моей модели я сделал разделение вручную, используя split_folders .

Затем я ImageDataGenerator определял, какой из них является поездом, проверкой и тестом.

Вот мой код

 validation_datagen = ImageDataGenerator()  test_datagen = ImageDataGenerator()  train_generator = train_datagen.flow_from_directory(train_dir)  validation_generator = validation_datagen.flow_from_directory(validation_dir)  test_generator = test_datagen.flow_from_directory(test_dir)  

когда я бегу

 confusion_matrix(test_generator, y_pred)  

с помощью y_pred

 y_pred = model.predict(test_generator)  

я получил ошибку, что мои входные переменные с несистематическим количеством выборок.

Что мне нужно сделать, чтобы образец был последовательным?