【发布时间】:2017-03-15 23:15:03
【问题描述】:
我有以下代码:
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 张图片。
【问题讨论】:
-
问题的 C++ 部分是什么?
-
Nothing...我已经删除了标签并添加了一些相关标签。
-
我假设您在代码中定义了
(X_valid, Y_valid)。如果是这样,请检查您是否可以使用ImageDataGenerator和train_on_batch()进行训练。如果可以,那么至少模型中没有错误。
标签: python deep-learning keras conv-neural-network