画loss、acc曲线

VIP文章 hamburgerJ

已于 2022-07-09 23:14:05 修改

阅读量1.1k

点赞数 1

分类专栏： pytorch 文章标签：人工智能

于 2022-07-09 23:13:27 首次发布

本文链接：https://blog.csdn.net/hamburgerJ/article/details/125700379

版权

# -*-coding:utf-8-*-
from matplotlib import pyplot as plt
import numpy as np


def ReadData(data_loc):
    epoch_list = []
    train_loss_list = []
    test_accuracy_list = []

    # open(data_loc,"r").readlines()
    with open(data_loc, "r") as f:
        linedata = f.readlines()

        for line_i in linedata:
            data = line_i.split(' ')
            print("data = ", data)
            epoch_i, train_loss_i, test_accuracy_i = data[1][0], data[3], data[6]
            epoch_list.append(int(epoch_i))
            train_loss_list.append(float(train_loss_i))
            test_accuracy_list.append(float(test_accuracy_i))

    # print(epoch_list)
    # print(train_loss_list)
    # print(test_loss_list)
    # print(test_accuracy_list)
    return epoch_list, train_loss_list,test_accuracy_list


def DrawLoss(train_loss_list, train_loss_list_2):
    plt.style.use('seaborn-ticks')
    # 更