python 的 csv 导入功能配上曲线图的可视化能力,做训练结果还挺方便的。你只需要把训练日志存在.csv
文件里,再用几行代码就能画出一张清晰的 loss 曲线,数据点不是整数也不影响,画出来一样顺滑。
支持在同一个图里画多条线,像训练集和验证集 loss 对比、不同模型的表现差异,一张图就能一目了然。还可以自定义图的标题、x 轴、y 轴标签,方便标记你关心的指标,比如epoch
、accuracy
啥的。
这个资源比较适合你在训练新模型、调参的时候快速检查效果。嗯,还有个小建议:尽量别用 Excel 保存csv
,有时候会改了编码格式,python 读起来出错。你可以用pandas.read_csv()
,响应也快,代码也简单。
如果你还在用 matlab 画图,那也没问题,下面这些链接也挺有参考价值的,比如matlab 的拟合曲线图,或者带误差阴影的二维图,场景不同选用就行。