plot_epoch_graph.py

# coding: utf-8

# In[1]:

import matplotlib.pyplot as plt


# In[2]:

epoch_range = [i for i in range(10)]


# In[12]:

avgLoss = [0.16270648973265425,
 0.078167325279861505,
 0.056490517006928372,
 0.045754109328929919,
 0.039341966316706355,
 0.034869728095849629,
 0.031900029562063045,
 0.02960749039253411,
 0.027881439287365618,
 0.026373543165770397]


# In[13]:

avgLoss


# In[14]:

vLoss_mat = [[0.086034561495934359, 0.97150155079775846],
 [0.05846608344584777, 0.98180146390581369],
 [0.048632269660400899, 0.98547794056937021],
 [0.043705952958791087, 0.98740592253912851],
 [0.040718987807195649, 0.98841132179563274],
 [0.038850192083252293, 0.98933896121426756],
 [0.03799936801437151, 0.98955522932736162],
 [0.036989176148385858, 0.99008693982814922],
 [0.036810977467315119, 0.99031106327554863],
 [0.036064139247890287, 0.99071221464666825]]
vLoss = [a[0] for a in vLoss_mat]


# In[15]:

vLoss


# In[16]:

plt.plot(epoch_range,avgLoss,'r--',epoch_range,vLoss,'b--')


# In[17]:

plt.show()