每当我使用fit_generator()时,我都会收到以下警告:
主要的
.py:5:UserWarning:Keras 2参数的语义
steps_per_epoch
与Keras 1参数不同
samples_per_epoch
.
是每个历元从生成器中提取的批次数。基本上,steps\u per\u epoch=samples\u per\u epoch/batch\u size。类似地
nb_val_samples
-&燃气轮机;
validation_steps
和
val_samples
-&燃气轮机;
steps
争论已经改变。相应地更新方法调用。
主要的
.py:5:用户警告:更新您的
fit_generator
调用Keras 2 API:
fit_generator(<keras.pre..., steps_per_epoch=60000, validation_data=<keras.pre..., epochs=1, validation_steps=10000)
以下是我的模型代码(简单的MNIST线性分类器,但我使用的每个模型都会收到此警告):
model = Sequential([
Lambda(normalize_input, input_shape=(1, 28, 28)),
Flatten(),
Dense(10, activation='softmax')
])
model.compile(Adam(),
loss='categorical_crossentropy',
metrics=['accuracy'])
model.fit_generator(batches,
steps_per_epoch=steps_per_epoch,
nb_epoch=1,
validation_data=test_batches,
nb_val_samples=test_batches.n)
我明白这个警告告诉我什么。这对我来说不是问题。我怎样才能摆脱它?