def epoch_end(self, l: Learner):
data_in, data_out = self.ds[:]
if l.categories:
split_in, split_out = l.split_data(data_in, data_out, split=2737)
model_pred = [l.model(item) if len(item)>0 else {'P(A|B)':l.TensorNaN(device=data_in.device)} for item in split_in]
l.loss_fn(model_pred, split_out, l, self.recorder, weighted=self.weighted, categories=True)
else:
output = l.model(data_in)
l.loss_fn(output, data_out, l, self.recorder, weighted=self.weighted) # this logs the data to the recorder
