我有以下代码:
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] 删除。
我来说两句