【发布时间】:2019-10-20 15:22:57
【问题描述】:
从this tutorial 代码,我正在尝试使用 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()
任何建议将不胜感激。
【问题讨论】:
-
您的超链接没有链接到教程。
-
@MigB 此代码是 'graphsage-cora-example.py',GraphSAGE Cora 节点分类示例。您可以在该链接中找到它。
标签: python machine-learning keras neural-network conv-neural-network