2017-03-15 4 views
3

Ich habe den folgenden Code:Keras Modell fit_generator ungerade Fehler

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)) 

Welche diese ungewöhnliche Fehler wirft

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) 

Kann jemand erklären, was hier schief geht, ich in einem Satz von 3700 bin Laden Bilder.

+0

Was ist der C++ Teil der Frage? –

+0

Nichts ... Ich habe das Tag entfernt und einige relevante Tags hinzugefügt. –

+0

Ich gehe davon aus, dass Sie '(X_valid, Y_valid)' im Code definiert haben. Wenn ja, prüfen Sie, ob Sie mit ImageDataGenerator und train_on_batch() trainieren können. Wenn Sie könnten, dann gibt es zumindest keinen Fehler im Modell. –

Antwort

3

Es könnte von der neuen API kommen (Keras 2.0 gestern veröffentlicht), die fit_generator() dauert jetzt steps_per_epoch Argument anstelle von samples_per_epoch.

Die steps_per_epoch ist typischerweise samples_per_epoch/batch_size.

Sie finden diese Information in the documentation.

Hilft es?

+0

Nein. Ich aktualisiere Keras aber immer noch diese Ausnahme –

+0

Sie könnten ein wenig genauer auf den Code sein, den Sie qnd Version Ihrer Software verwenden. –

+0

ich benutze Keras 2.0.8 und hier ist code: https://gist.github.com/creotiv/c7fb0328de22ec021f405ebbdc202100 –

Verwandte Themen