Keras fit_generator () не показывает точность на Colab - PullRequest
0 голосов
/ 20 октября 2019

Я только что перешел со своего локального компьютера, чтобы обучить модель на Colab.

Я думаю, что может быть проблема с Colab, так как Keras fit_generator () не сообщает мне точность (только потеря). Как сделать показанную точность?

Код может отображать точность на моем ПК. Однако код на 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

1 Ответ

1 голос
/ 20 октября 2019

Это означает, что вы не указали никаких метрик при построении модели, в строке model.compile вы должны указать метрику accuracy, чтобы она отображалась во время обучения:

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