[英]Keras model fit_generator odd error
我有以下代碼:
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(昨天發布的fit_generator()
2.0), fit_generator()
現在采用steps_per_epoch
參數而不是samples_per_epoch
。
steps_per_epoch
通常是samples_per_epoch
/ batch_size
。
您可以在文檔中找到此信息。
有幫助嗎?
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.