python - NotImplementedError:尚未为未构建的模型子类启用`fit_generator`
问题描述
我正在使用以下代码:
import tensorflow as tf
##############################################################
traindata = tf.keras.preprocessing.image.ImageDataGenerator(
rescale=1. / 255,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True)
input = traindata.flow_from_directory('VS/train')
modelo = tf.keras.Sequential()
modelo.add(tf.keras.layers.Conv2D(32, (3, 3),
activation=tf.keras.activations.relu))
modelo.add(tf.keras.layers.Flatten())
modelo.add(tf.keras.layers.Dense(64, activation=tf.keras.activations.relu))
modelo.add(tf.keras.layers.Dense(2, activation=tf.keras.activations.relu))
modelo.compile(loss='categorical_crossentropy', optimizer='rmsprop')
modelo.fit_generator(input, epochs=1)
但是,我收到此错误:
所以通过运行下面的代码,我得到了这个错误
NotImplementedError: `fit_generator` is not yet enabled for unbuilt Model subclasses
有人可以告诉我有什么问题吗?
解决方案
您没有input_shape
在第一层指定,所以模型没有完全定义。此过程尚未使用 实现fit_generator
,因此您应该使用初始完全定义模型input_shape
。
推荐阅读
- powerbi - Power BI 总计评估
- java - 关闭在 CMD 上运行的 Spring MVC 应用程序
- python-3.x - 如何使用 pandas 编辑 .CSV 文件中的列
- javascript - 显示标题 - 在手机上弹跳的问题
- node.js - 差异加载在 Angular 12 中不起作用:“Uncaught SyntaxError: Unexpected token ':'”
- mysql - 使用公用表表达式时 JOIN 出错
- docker - 如何使用 apparmor 在 docker 容器中启用挂载文件系统
- assembly - emu 8086 Keep symbols unchanged
- c++ - Array 打印随机符号
- sql - Laravel 查询构建器从子查询左连接