fit_generator() got an unexpected keyword argument 'nb_epoch' というエラーが出て困っています。

Google Colabを使って機械学習のプログラムを書いています。
少し前まで、問題なく動いていたプログラムを最近動かしてみると、ファインチューニングの部分で
"fit_generator() got an unexpected keyword argument 'nb_epoch'"というエラーが表示されてしまい、
訓練できなくなりました。

以前には普通に動いており、それからどこもいじっていないので、keras等のアップデートが原因ではないかと思っているのですが、
調べても上手く解決できません。

解決策をご存知の方がいましたら、アドバイス頂けると幸いです。
宜しくお願い致します。

meg_

2020/08/20 16:03

> 少し前まで、問題なく動いていたプログラム上記のときと現在とで使用しているモジュールのバージョンが変わっていませんか？以前のバージョンにダウンすれば良いでしょう。

Charlotte78

2020/08/25 06:58

アドバイスして頂き有難うございました. kerasとtensorflowを以前のverにダウングレードする事で解決致しました.

行動規範の内容に同意します

回答1件

ベストアンサー

The nb_epoch argument has been renamed epochs everywhere. とのことです。
https://github.com/keras-team/keras/wiki/Keras-2.0-release-notes

fit_generatorの仕様

python
1fit_generator(generator, steps_per_epoch=None, epochs=1,
2  verbose=1,callbacks=None, validation_data=None,
3  validation_steps=None, class_weight=None, max_queue_size=10,
4  workers=1, use_multiprocessing=False, shuffle=True, initial_epoch=0)

関連するトピックが既にあるようです。
keras, Gridsearch, エポック1回しか出来てない？

プログラムに少しだけ手を入れることをいとわないのでしたら、バージョンアップによって整合がとれなくなっている部分を調整してみてはいかがでしょうか？

投稿2020/08/21 16:39

PINTO

総合スコア351

Charlotte78

2020/08/25 06:57

ご回答頂き有難うございました. keras2.0からnb_epochをepochに変更する必要がありましたので,変更しました所, Fine-tuning時に表示されていたepoch毎のval_lossとval_accが表示されなくなってしまったので, kerasとtensorflowを以前のverにダウングレードする事で解決致しました.

行動規範の内容に同意します