(self)
| 160 | return self.loss,self.loss_gradient |
| 161 | |
| 162 | def plot_loss(self): |
| 163 | if self.ax_loss.lines: |
| 164 | self.ax_loss.lines.remove(self.ax_loss.lines[0]) |
| 165 | self.ax_loss.plot(self.train_mse, 'r-') |
| 166 | plt.ion() |
| 167 | plt.xlabel('step') |
| 168 | plt.ylabel('loss') |
| 169 | plt.show() |
| 170 | plt.pause(0.1) |
| 171 | |
| 172 | |
| 173 |