MLPRegressor 的损失历史记录

Mar*_*ren 5 python scikit-learn

我正在使用 MLPRegressor 来解决问题,并且想绘制损失函数,即每个训练时期损失减少了多少。但是,可用于 MLPRegressor 的属性model.loss_仅允许访问最后的损失值。是否有可能访问整个损失历史记录?

Fra*_*ggi 4

实际上非常简单:model.loss_curve_给出每个时期的损失值。然后,您可以通过将时期放在 x 轴上并将上述值放在 y 轴上来轻松绘制学习曲线

  • 请注意,这仅适用于默认的“adam”解算器。“sgd”或“lbfgs”求解器不会生成“model.loss_curve_”。我花了一段时间才弄清楚! (11认同)