keras - 将 model.fit_generator 转换为 model.fit
问题描述
我在下面有代码,
train_datagen = ImageDataGenerator(
rescale=1./255,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True)
test_datagen = ImageDataGenerator(rescale=1./255)
train_generator = train_datagen.flow_from_directory(
'data/train',
target_size=(150, 150),
batch_size=32,
class_mode='binary')
validation_generator = test_datagen.flow_from_directory(
'data/validation',
target_size=(150, 150),
batch_size=32,
class_mode='binary')
现在model.fit_generator
定义如下:
model.fit_generator(
train_generator,
steps_per_epoch=2000,
epochs=50,
validation_data=validation_generator,
validation_steps=800)
现在已弃用,在这种情况下model.fit_generator
更改为的正确方法是什么model.fit_generator
?model.fit
解决方案
您只需更改model.fit_generator()
为model.fit()
.
从 TensorFlow 2.1 开始,model.fit()
也接受生成器作为输入。就如此容易。
来自 TensorFlow 的官方文档:
警告:此功能已弃用。它将在未来的版本中删除。更新说明:请使用支持生成器的 Model.fit。
推荐阅读
- java - 测试涉及 ScheduledExecutorService#scheduleAtFixedRate 的代码的单元测试失败
- angular - Angular - 显示来自子组件内的导入组件的激活路由数据
- uwp - 在 InkCanvas 中调整已绘制的 Inkstrokes 的大小
- flutter - Flutter Widgets - 访问父流
- python - 如何在远程 linux 服务器中查看 python 代码的结果
- python - 基于三元条件更新数组的不同元素
- c# - 拒绝更改后的 Entity Framework Core dbContext 行为
- javascript - Webpack: How to inject process.env runtime on client(browser) side making build independent of environment
- php - PHP:将目录和子目录中的文件列出到 XML 文件中
- groovy - 将此典型字符串转换为詹金斯管道中的映射