plot_train_epoch_loss.py

from matplotlib import pyplot, dates
from csv import reader
from dateutil import parser
import pdb

with open('logs/epoch_loss_train.txt', 'r') as f:
    data = list(reader(f))

epoch_loss = [float(i[1]) for i in data]
epoch = [float(i[0]) for i in data]

pyplot.plot(epoch, epoch_loss)
pyplot.title('Epoch Loss Train')
pyplot.xlabel('Number of epochs')
pyplot.ylabel('Epoch loss')
pyplot.show()