# -*-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 = ()
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):
('seaborn-ticks')
# 更多style请看：/viviliving/article/details/107690844
("Loss")
("epoch")
("loss")
train_loss_list = train_loss_list[:10]
epoch_list = [i for i in range(len(train_loss_list))]
p1, = (epoch_list, train_loss_list, linewidth=3)
p2, = (epoch_list, train_loss_list_2, linewidth=3)
([p1, p2], ["with pretrain", "no pretrain"])
()
def DrawAcc(train_loss_list, train_loss_list_2):
('seaborn-ticks')
("Accuracy")
("epoch")
("accuracy")
train_loss_list = train_loss_list[:10]
epoch_list = [i for i in range(len(train_loss_list))]
p1, = (epoch_list, train_loss_list, linewidth=3)
p2, = (epoch_list, train_loss_list_2, linewidth=3)
([p1, p2], ["with pretrain", "no pretrain"])
()
if __name__ == '__main__':
data_1_loc = "loss_pre.txt"
data_2_loc = ""
_, train_loss_list, test_accuracy_list = ReadData(data_1_loc)
_, train_loss_list_2, test_accuracy_list_2 = ReadData(data_2_loc)
DrawLoss(train_loss_list, train_loss_list_2)
DrawAcc(test_accuracy_list, test_accuracy_list_2)