(self)
| 164 | return self.loss, self.loss_gradient |
| 165 | |
| 166 | def plot_loss(self): |
| 167 | if self.ax_loss.lines: |
| 168 | self.ax_loss.lines.remove(self.ax_loss.lines[0]) |
| 169 | self.ax_loss.plot(self.train_mse, "r-") |
| 170 | plt.ion() |
| 171 | plt.xlabel("step") |
| 172 | plt.ylabel("loss") |
| 173 | plt.show() |
| 174 | plt.pause(0.1) |
| 175 | |
| 176 | |
| 177 | def example(): |