Keras模型fit_generator奇数错误

山姆

我有以下代码:

    datagen = ImageDataGenerator(
        rescale=1./255,
        target_size=(128, 128),
        shear_range=0.2,
        zoom_range=0.2,
        horizontal_flip=True)

    test_datagen = ImageDataGenerator(
        rescale=1./255,
        target_size=(128, 128)
    )

    datagen.fit(X_train)

    model.fit_generator(
        datagen.flow(X_train, Y_train),
        samples_per_epoch=len(X_train),
        epochs=30,
        verbose=1,
        validation_data=(X_valid, Y_valid))

哪个抛出此异常错误

  Traceback (most recent call last):
      File "cnn.py", line 258, in <module>
          models = run_cross_validation_create_models(num_folds)
      File "cnn.py", line 205, in run_cross_validation_create_models
          validation_data=(X_valid, Y_valid))
      TypeError: fit_generator() takes at least 4 arguments (5 given)

有人可以解释这里出了什么问题吗,我正在加载3700张图像。

纳西姆一世

它可能来自新的API(昨天发布的Keras 2.0),fit_generator()现在使用steps_per_epoch参数代替samples_per_epoch

steps_per_epoch是典型地samples_per_epoch/ batch_size

您可以在文档中找到此信息

有帮助吗?

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

在Keras模型中使用fit_generator

为什么keras模型可以编译,但是fit_generator命令会抛出“模型未编译的运行时错误”?

在Keras使用fit_generator

Keras fit_generator问题

Keras fit_generator上的错误,运行了几步

对于Keras中的LSTM和fit_generator,出现错误“使用之前必须先编译模型”

使用 fit_generator 批量训练模型

在Keras中多次调用“ fit_generator()”

Keras fit_generator运行非常缓慢

fit_generator()在keras中的优势

Keras Fit_generator回调

Keras的fit_generator额外训练价值

在Keras中正确使用fit_generator

Keras fit_generator() 用于长信号

Python无法将fit_generator应用于具有多个输入的keras模型

使用 fit_generator 时,Keras 模型的批量大小为 1

使用fit_generator拟合keras模型时如何修复“MemoryError”?

Keras fit_generator:模型也使用validation_data学习吗?

将Keras Functional API与fit_generator一起使用时的输入形状错误

Keras fit_generator 使用输入和输出图像生成器“ndim”错误

Keras-SegNet使用ImageDataGenerator和fit或fit_generator

我冻结了所有的keras层,但是使用fit_generator时模型发生了变化

Keras:使用fit_generator时出现notImplementedError / RuntimeError

Keras fit_generator()-时间序列的批处理如何工作?

Keras fit_generator具有多个输入层

如何在Keras中使用fit_generator()加权类?

使用.fit_generator()在keras中训练GAN

Keras fit_generator和steps_per_epoch

使用fit_generator形状不匹配时出错(Keras)