multiml.task.pytorch.pytorch_metrics module

multiml.task.pytorch.pytorch_metrics.dummy(*args, **kwargs)

class multiml.task.pytorch.pytorch_metrics.EpochMetric(metrics, enable, ml)

Bases: object

Utility class to manage epoch metrics.

__init__(metrics, enable, ml)

loss(batch_result)

subloss(batch_result)

acc(batch_result)

lr(batch_result)

pred(batch_result)

all_preds()

multiml.task.pytorch.pytorch_metrics.get_pbar_metric(epoch_result)