令history = model.fit(...)，用history使得训练结果可视化，并在过拟合之前提前结束训练（tf，keras）

最新推荐文章于 2025-08-21 19:47:57 发布

原创

最新推荐文章于 2025-08-21 19:47:57 发布 · 1.3w 阅读

52 ·

CC 4.0 BY-SA版权

文章标签：

#可视化 #tensorflow #深度学习

这段代码展示了如何在Keras中构建模型并利用EarlyStopping回调进行训练优化。通过监控验证集损失，当验证损失在10个连续的epoch内没有改善时，训练将提前停止。同时，定义了一个辅助函数`plot_history`来绘制平均绝对误差和均方误差随训练进程的变化，帮助理解模型的训练效果。

hist = pd.DataFrame(history.history)
hist['epoch'] = history.epoch
hist.tail()

def plot_history(history):
  hist = pd.DataFrame(history.history)
  hist['epoch'] = history.epoch

  plt.figure()
  plt.xlabel('Epoch')
  plt.ylabel('Mean Abs Error [MPG]')
  plt.plot(hist['epoch'], hist['mae'],
           label='Train Error')
  plt.plot(hist['epoch'], hist['val_mae'],
           label = 'Val Error')
  plt.ylim([0,5])
  plt.legend()

  plt.figure()
  plt.xlabel('Epoch')
  plt.ylabel('Mean Square Error [$MPG^2$]')
  plt.plot(hist['epoch'], hist['mse'],
           label='Train Error')
  plt.plot(hist['epoch'], hist['val_mse'],
           label = 'Val Error')
  plt.ylim([0,20])
  plt.legend()
  plt.show()


plot_history(history)