我刚从本地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
答案 0 :(得分:1)
这意味着您没有在模型构建过程中指定任何度量标准,在model.compile
行中,您必须指定accuracy
度量标准才能在训练期间显示:
model.compile(loss=..., optimizer=..., metrics=["accuracy"])