tensorflow2.0 神经网络多次训练结果的可视化

针对自己的数据集,我们用不同的网络来训练,或者用同一网络来测试不同的数据集,都会得到很多个训练结果。为了在此之间进行对比,我们最好把所有的结果绘制在一张图中。下面我们来提供一种方法。
我们在训练网络的过程中会有训练日志输出,针对每一个epoch,都会有训练准确率acc和训练损失loss,还有验证准确率val_acc和验证损失val_loss,还有最后的测试准确率test_acc。我们需要把他们写入一个字典当中,进行调用。然后可视化,接下来上代码:

import matplotlib.pyplot as plt

#用不同的字典将不同的训练结果写入
dict1={
   '_4_loss':[2.9492,1.4026,0.9703,0.8491,0.8409,0.6915,0.6087,0.6116,0.4860,0.5437,0.4484,0.4296,0.4486,0.4082,0.3752,0.4040,0.3312,0.3715,0.2840,0.3193,0.3025,0.3879,0.3137,0.3200,0.2683,0.2574,0.2987,0.3124,0.2497,0.3511,0.2578,0.2828,0.3018,0.3333,0.2968,0.4256,0.3496,0.3688,0.3829,0.2439,0.2508,0.2581,0.2181,0.2557,0.2523,0.3288,0.2883,0.2523,0.3114,0.3476],
       '_4_acc':[0.3090,0.5401,0.6557,0.7123,0.7311,0.7571,0.7854,0.7783,0.8302,0.8113,0.8561,0.8443,0.8467,0.8608,0.8703,0.8656,0.8679,0.8821,0.8939,0.8892,0.9033,0.8797,0.8821,0.8986,0.8939,0.9033,0.9057,0.9080,0.9245,0.8892,0.9104,0.9080,0.8915,0.8962,0.8962,0.8750,0.9033,0.8797,0.8939,0.9104,0.9057,0.9127,0.9269,0.9198,0.9057,0.9009,0.8939,0.9080,0.8915,0.8844],
       '_4_val_loss':[19.5177,8.5774,10.5291,6.7281,5.6474,7.3317,5.1849,9.2877,13.5350,5.4513,3.3073,7.0256,8.5327,5.3727,6.9400,4.3825,2.7396,2.6096,3.5402,4.3104,3.3585,3.0279,4.6128,3.9091,2.9176,2.2243,2.6610,2.8934,2.0937,2.2669,3.5162,3.2038,2.1915,2.7525
  • 4
    点赞
  • 27
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值