5.1. Visualizing our cost

Summary of our cost after 400 Epochs

In [20]:
# Ploting our cost
plt.plot(loss_values)
plt.title('Cross Entropy Loss')
plt.ylabel('Loss')
plt.xlabel('Epochs')
Out[20]:
Text(0.5,0,'Epochs')