Keras fit_generator()在Colab上不显示准确性

时间:2019-10-20 11:30:49

标签: keras deep-learning google-colaboratory

我刚从本地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)

我希望结果也显示为acc和loss_acc,例如:

Epoch 1/10
 - 48s - loss: 13.6619 - val_loss: 13.9256
Epoch 2/10
 - 48s - loss: 13.6741 - val_loss: 13.7124

1 个答案:

答案 0 :(得分:1)

这意味着您没有在模型构建过程中指定任何度量标准,在model.compile行中,您必须指定accuracy度量标准才能在训练期间显示:

model.compile(loss=..., optimizer=..., metrics=["accuracy"])