我应该如何分割Keras数据生成器train-valid-test
中的数据集?我应该将train文件夹作为train有效子集加载,并加载test文件夹和测试以与model.predict(...).
一起使用还是再次使用valid_generator
?
train_datagen = ImageDataGenerator(rescale=1./255)
valid_datagen = ImageDataGenerator(rescale=1./255)
train_generator = train_datagen.flow_from_directory(
train_dir,
target_size=(224, 224),
class_mode='categorical')
valid_generator = valid_datagen.flow_from_directory(
test_dir,
target_size=(224, 224),
class_mode='categorical')
history = model.fit(train_generator,
epochs=10,
validation_data=valid_generator,
verbose=1)
,这里是预测步骤
pred = model.predict(valid_generator)
1条答案
按热度按时间3pmvbmvn1#
由于训练数据集的数据扩充需求,通常需要
ImageDataGenerator
的多个示例,而没有扩充的相同ImageDataGenerator
可用于验证和测试数据集(以及它们对应的目录路径)。在训练模型之后,model.evaluate_generator
可用于评估测试数据集上的模型性能。以下是给定模型和数据集有效目录的简单示例: