Added axes label to the plot

Author: P-Shreyas-Shetty

neural_network/bpnn.py

@@ -164,6 +164,8 @@ def plot_loss(self):
             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)