INNER CODE UNIT · Python
update_graph
timeseriesAI/tsai · tsai/callback/core.py:105
def update_graph(self, graphs, x_bounds=None, y_bounds=None, figsize=(6,4)):
if not hasattr(self, 'graph_fig'):
self.graph_fig, self.graph_ax = plt.subplots(1, figsize=figsize)
self.graph_out = display(self.graph_ax.figure, display_id=True)
self.graph_ax.clear()
if len(self.names) < len(graphs): self.names += [''] * (len(graphs) - len(self.names))
for g,n in zip(graphs,self.names):
if (g[1] == [None] * len(g[1])): continue
self.graph_ax.plot(*g, label=n)
self.graph_ax.legend(loc='upper right')
self.graph_ax.grid(color='gainsboro', linewidth=.5)
if x_bounds is not None: self.graph_ax.set_xlim(*x_bounds)
if y_bounds is not None: self.graph_ax.set_ylim(*y_bounds)
self.graph_ax.set_title(f'Losses\nepoch: {self.epoch +1}/{self.n_epoch}')
self.graph_out.update(self.graph_ax.figure)
ShowGraphCallback2 = ShowGraph