2017-09-20 13 views
2

私はJupyterノートブックでPython 3カーネルでKeras 2.0.8を使用しています。私のバックエンドはTensorFlow 1.3で、私はMacで開発しています。Keras 2 fit_generator UserWarning: `steps_per_epoch`はKeras1引数` samples_per_epoch`と同じではありません

私は(fit_generatorを使用していますたび)私は次の警告取得しています:

/Users/username/anaconda/envs/tensorflow/lib/python3.6/site-packages/ipykernel/ メイン .py:5:UserWarning:Keras 2引数steps_per_epochのセマンティクスは、Keras 1引数samples_per_epochと同じではありません。 steps_per_epochは、各エポックでジェネレータから描画するバッチの数です。基本的にはsteps_per_epoch = samples_per_epoch/batch_sizeです。同様にnb_val_samples - >validation_stepsval_samples - >steps引数が変更されました。それに応じてメソッド呼び出しを更新します。 /Users/username/anaconda/envs/tensorflow/lib/python3.6/site-packages/ipykernel/ メインの.py:5:UserWarning:fit_generator(<keras.pre..., steps_per_epoch=60000, validation_data=<keras.pre..., epochs=1, validation_steps=10000)

:Keras 2 APIへのご fit_generator呼び出しを更新以下

は私のモデルのコードです(簡単なMNIST線形分類器が、私は私が使用するすべてのモデルのために、この警告を取得しています):

model = Sequential([ 
    Lambda(normalize_input, input_shape=(1, 28, 28)), 
    Flatten(), 
    Dense(10, activation='softmax') 
]) 
model.compile(Adam(), 
       loss='categorical_crossentropy', 
       metrics=['accuracy']) 

そして、この私のfit_generatorである()の呼び出し:

model.fit_generator(batches, 
       steps_per_epoch=steps_per_epoch, 
       nb_epoch=1, 
       validation_data=test_batches, 
       nb_val_samples=test_batches.n) 

私はこの警告が何を伝えているのか理解しています。私の場合、これは問題ではありません。どうすればそれを取り除くことができますか?

答えて

3

この警告は、関数呼び出しにKeras 1.0キーワードがある場合に発生します。 nb_epochepochsに、nb_val_samplesvalidation_stepsに置き換えて、関数呼び出しを更新してください。

+0

ありがとうございました!どういうわけか私はそれを逃した。 –

関連する問題