def plot_training(history=None, metric='acc', title='Model Accuracy', loc='lower right'): model_list = [] fig = plt.figure(figsize=(10, 8)) for key, val in history.items(): model_list.append(key.replace(HISTORY_DIR, '').rstrip('.history')) plt.plot(val[metric]) plt.title(title) plt.ylabel(metric) plt.xlabel('epoch') plt.legend(model_list, loc=loc) plt.show()
时间: 2024-03-31 11:33:25 浏览: 113
plot_keras_history:一个简单的python程序包,用于打印keras NN训练历史记录
这段代码是一个用于绘制模型训练历史数据的函数。它接受一个历史数据字典,用于绘制每个模型在训练过程中的准确率或损失率的变化。metric参数指定绘制的指标,可以是"acc"(准确率)或"loss"(损失率),title参数指定图表标题,loc参数指定图例的位置。在函数中,首先创建一个画布,然后遍历历史数据字典中的每个模型,将其准确率或损失率绘制到图表中。最后展示图表。
阅读全文