mixmo.core.metrics_wrapper.MetricsWrapper

class mixmo.core.metrics_wrapper.MetricsWrapper(metrics)

Bases: object

Metric storing object

__init__(metrics)

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(metrics)	Initialize self.
get_scores([split])	Print stored results
update(pred, target, confidence, probs)	Compute tracked metrics and update records
update_diversity(target, predictions)	Compute and update records of diversity metrics

_get_diversity_stats()

Retrieve stored diversity stats

get_scores(split='train')

Print stored results

update(pred, target, confidence, probs)

Compute tracked metrics and update records

update_diversity(target, predictions)

Compute and update records of diversity metrics