我刚从本地PC移到Colab上训练模型。
我认为Colab可能存在问题,因为Keras fit_generator()不会告诉我准确性(只是损失)。如何使显示的准确性?
该代码可以在我的PC上显示准确性。但是,Colab上的代码如下所示
# Model Compile
model.compile(optimizer='Adam', loss='categorical_crossentropy')
# Training
nb_epochs = 10
model.fit_generator(
train_generator,
steps_per_epoch = train_generator.samples // batch_size,
validation_data = validation_generator,
validation_steps = validation_generator.samples // batch_size,
epochs = nb_epochs
verbose = 2)
结果没有准确性报告。
Epoch 1/10
- 48s - loss: 13.6619 - val_loss: 13.9256
Epoch 2/10
- 48s - loss: 13.6741 - val_loss: 13.7124
我希望结果也显示为acc和loss_acc,例如:
Epoch 1/50
- 791s - loss: 1.1517 - acc: 0.6122 - val_loss: 1.2601 - val_acc: 0.5735
这意味着您没有在模型构建期间指定任何度量标准,而在这一model.compile
行中,您必须指定accuracy
要在训练期间显示的度量标准:
model.compile(loss=..., optimizer=..., metrics=["accuracy"])
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句