From 74a65017ca114a1e070d45e8a2c3cf59d4391b8f Mon Sep 17 00:00:00 2001
From: P-Shreyas-Shetty <ps2shreyas@gmail.com>
Date: Tue, 13 Nov 2018 05:27:26 +0530
Subject: [PATCH] Added axes label to the plot

---
 neural_network/bpnn.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/neural_network/bpnn.py b/neural_network/bpnn.py
index 0865e35f0..92deaee19 100644
--- a/neural_network/bpnn.py
+++ b/neural_network/bpnn.py
@@ -164,6 +164,8 @@ class BPNN():
             self.ax_loss.lines.remove(self.ax_loss.lines[0])
         self.ax_loss.plot(self.train_mse, 'r-')
         plt.ion()
+        plt.xlabel('step')
+        plt.ylabel('loss')
         plt.show()
         plt.pause(0.1)