model_name : Epoch{i}-Total_Loss{m}-Val_Loss{n}.pth to figure

import matplotlib.pyplot as plt
import glob
import os
import numpy as np
import pylab as pl



file_names = os.listdir("824_4classes_200epoch_logs")

train_Loss_merge = np.empty(len(file_names))
val_Loss_merge = np.empty(len(file_names))
Epoch_merge = np.arange(1,202)


for file_name in file_names:
    #get epoch
    Epoch_, split_, _ = file_name.split('-')
    Epoch = Epoch_[5:]
    #get Loss
    train_Loss_index = file_name.index("Total_Loss")+10
    val_Loss_index = file_name.index("Val_Loss")+8
    train_Loss_index_dot = file_name.index('.')
    val_Loss_index_dot = file_name.index('.',train_Loss_index_dot+1)
    train_Loss = file_name[train_Loss_index:train_Loss_index_dot+3]
    val_Loss = file_name[val_Loss_index:val_Loss_index_dot+3]
    train_Loss_merge[int(Epoch)-1] = train_Loss
    val_Loss_merge[int(Epoch)-1] = val_Loss
    # merge[int(Epoch)-1] = np.array([Epoch,train_Loss,val_Loss])

# print(merge[:3])


plt.plot(Epoch_merge[3:],train_Loss_merge[3:],'g',label = u'Train Loss')
pl.legend()
plt.plot(Epoch_merge[3:],val_Loss_merge[3:],'r',label = u'Val Loss')
pl.legend()
pl.xlabel(u'Epoch')
pl.ylabel(u'Loss')
plt.title('200 Epoch for 4classes - 8.24')

plt.show()