как вызвать данные проверки из ImageDataGenerator кераса

Question

Я изучаю набор данных MNIST и решил попробовать перебор данных. Теперь напрямую:

Я генерирую данные, используя разделение проверки следующим образом:

datagen = keras.preprocessing.image.ImageDataGenerator(... , validation_split=0.21, ...)

datagen.fit(train_x)

, но я не знаю, как горячо назвать это разделение проверки на соответствие модели

hist = model.fit_generator(datagen.flow(train_x, train_y, batch_size =32), 
                                steps_per_epoch=len(train_x)//32, 
                                epochs = 70, verbose=0, callbacks= [PlotLossesKeras()], 
                                validation_data= **???**, <-----
                                validation_steps=None, 
                                validation_freq=1, 
                                class_weight=None, 
                                max_queue_size=10, 
                                workers=1, 
                                use_multiprocessing=False, 
                                shuffle=True,
                                initial_epoch=0)

Я не смог найти ответ ни в документации, ни где-либо еще. Вы можете мне помочь?

Answer 1

Вы должны указать, какое подмножество вы используете: subset = 'validation', поскольку в документации Keras указано:

подмножество: Подмножество данных ("обучение" или "проверка") ) если validation_split установлен в ImageDataGenerator.

Так, например, вы можете сделать следующее:

datagen = keras.preprocessing.image.ImageDataGenerator(..., validation_split=0.21)

train_generator = datagen.flow(..., subset='training')
valid_generator = datagen.flow(..., subset='validation')

hist = model.fit_generator(...,
        generator = train_generator,
        validation_data = valid_generator,
        steps_per_epoch = len(train_generator),
        validation_steps = len(valid_generator),
)

Вы также можете увидеть некоторые примеры, предоставленные самим Keras, верно после ImageDataGenerator class документация