未实现错误:`fit_generator` 对于未构建的模型子类尚未启用。

4
我将使用以下代码:

我正在使用以下代码:

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 

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

1个回答

7

您在第一层没有指定input_shape,因此模型未完全定义。这个过程没有与fit_generator实现,所以您应该使用初始的input_shape来完全定义模型。


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接