小编Tim*_* AI的帖子

NotImplementedError:尚未为未构建的Model子类启用`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)
Run Code Online (Sandbox Code Playgroud)

但是,我收到此错误:

所以通过运行下面的代码,我得到了这个错误

NotImplementedError: `fit_generator` is not yet enabled for unbuilt Model subclasses 
Run Code Online (Sandbox Code Playgroud)

有人能告诉我什么是错的吗?

python generator keras tensorflow

3
推荐指数
1
解决办法
1536
查看次数

标签 统计

generator ×1

keras ×1

python ×1

tensorflow ×1