Draw loss, acc curve

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')

# For more styles, please see: /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)