zinating
diff --git a/‎neural_network/bpnn.py‎
Lines changed: 2 additions & 0 deletions b/‎neural_network/bpnn.py‎
Lines changed: 2 additions & 0 deletions
@@ -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)