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)