NotImplementError: 尚未为未构建的模型子类启用"fit_generator"

Question

我使用以下代码：

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 

有人可以告诉我出了什么问题吗？

Answer 1

您没有在第一层中指定input_shape，因此模型未完全定义。此过程尚未在fit_generator中实现，因此您应该使用初始input_shape完全定义模型。