INNER CODE UNIT · Python
__init__
timeseriesAI/tsai · tsai/callback/core.py:64
def __init__(self, plot_metrics:bool=True, final_losses:bool=True, perc:float=.5):
store_attr()
def before_fit(self):
self.run = not hasattr(self.learn, 'lr_finder') and not hasattr(self, "gather_preds")
if not(self.run): return
self.nb_batches = []
self.learn.recorder.loss_idxs = [i for i,n in enumerate(self.learn.recorder.metric_names[1:-1]) if 'loss' in n]
_metrics_info = [(i,n) for i,n in enumerate(self.learn.recorder.metric_names[1:-1]) if 'loss' not in n]
if len(_metrics_info) > 0:
self.metrics_idxs, self.metrics_names = list(zip(*_metrics_info))
else:
self.metrics_idxs, self.metrics_names = None, None
def after_train(self): self.nb_batches.append(self.train_iter - 1)
def after_epoch(self):
"Plot validation loss in the pbar graph"