Как построить историю модели - PullRequest
0 голосов
/ 20 октября 2019

Из этого урока код, я пытаюсь построить историю модели, используя библиотеку keras. Но это дает ошибку KeyError:'acc'.

Это часть кода.

model.compile(
    optimizer=optimizers.Adam(lr=learning_rate, decay=0.001),
    loss=losses.categorical_crossentropy,
    metrics=[metrics.categorical_accuracy],
)

# Train model
history = model.fit_generator(
    train_gen, epochs=num_epochs, validation_data=val_gen, verbose=2, shuffle=False
)

plt.plot(history.history['acc'])
plt.plot(history.history['val_acc'])
plt.title('Model accuracy')
plt.ylabel('Accuracy')
plt.xlabel('Epoch')
plt.legend(['Train', 'Test'], loc='upper left')
plt.show()

# Plot training & validation loss values
plt.plot(history.history['loss'])
plt.plot(history.history['val_loss'])
plt.title('Model loss')
plt.ylabel('Loss')
plt.xlabel('Epoch')
plt.legend(['Train', 'Test'], loc='upper left')
plt.show()

Любые советы будут оценены.

1 Ответ

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

Измените acc на accuracy и val_acc на val_accuracy. и рассмотрим эту ссылку

...