如何用python画出训练曲线图

import pickle
import matplotlib.pyplot as plt
f1 = open('/media/xm/0ABA09D10ABA09D1/交通标志识别/Capsule-master/cnn/traffic_cnn_show.txt','rb')
cnn = pickle.load(f1)
f2 = open('/media/xm/0ABA09D10ABA09D1/交通标志识别/Capsule-master/capsule/traffic_capsule_show.txt','rb')
capsule = pickle.load(f2)
def loss_plot(loss_type):
        iters = range(len(cnn['self.losses']))
        plt.figure()
        # acc
        #plt.plot(iters, self.accuracy[loss_type], 'r', label='train acc')
        # loss
        plt.plot(iters, cnn['self.val_loss'], 'g', label='cnn_val_loss')
        if loss_type == 'epoch':
            # val_acc
            #plt.plot(iters, self.val_acc[loss_type], 'b', label='val acc')
            # val_loss
            plt.plot(iters, capsule['self.val_loss'], 'k', label='capsule_val_loss')
        plt.grid(False)
        plt.xlabel(loss_type)
        plt.ylabel('val_loss')
        plt.legend(loc="upper right")
        plt.show()
        plt.figure()
        # acc
        m = []
        n = []
        for x in cnn['self.val_acc']:
                m.append(x)
        for y in capsule['self.val_acc']:
                n.append(y)
        
        plt.plot(iters, m, 'r', label='cnn_val_acc')
        # loss
        #plt.plot(iters, self.losses[loss_type], 'g', label='train loss')
        if loss_type == 'epoch':
            # val_acc
            plt.plot(iters, n, 'b', label='capsule_val_acc')
            # val_loss
            #plt.plot(iters, self.val_loss[loss_type], 'k', label='val loss')
        plt.grid(False)
        plt.xlabel(loss_type)
        plt.ylabel('val_accuracy')
        plt.legend(loc="lower right")
        plt.show()
loss_plot('epoch')

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值