bugfix> python > 投稿

私は次のコードを使用しています:

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 件
  • input_shape を指定しませんでした  そのため、モデルは完全には定義されていません。このプロセスは fit_generator では実装されていません 、したがって、最初の input_shape でモデルを完全に定義する必要があります 。

あなたの答え