class monad.config.training_params.CustomMetric

Defines a custom torchmetrics.Metric.

from monad.config.training_params import CustomMetric
from torchmetrics.classification import F1Score

CustomMetric(alias="f1", metric=F1Score(task="binary"))

alias: str
Name assigned by the user to identify metric.

metric_name: torchmetrics.Metric
Instance of the torchmetrics.Metric