Added axes label to the plot

2018-11-13 05:27:26 +05:30
parent 70a6d98e0f
commit 74a65017ca
1 changed files with 2 additions and 0 deletions
--- 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)