diff --git a/fuse/eval/metrics/classification/metrics_classification_common.py b/fuse/eval/metrics/classification/metrics_classification_common.py index 33abb01b..61ea008b 100644 --- a/fuse/eval/metrics/classification/metrics_classification_common.py +++ b/fuse/eval/metrics/classification/metrics_classification_common.py @@ -353,6 +353,8 @@ def __init__( ): """ See MetricDefault for the missing params + :param pred: class label predictions + :param target: ground truth labels :param sample_weight: weight per sample for the final accuracy score. Keep None if not required. """ super().__init__( @@ -392,6 +394,8 @@ def __init__( ): """ See MetricDefault for the missing params + :param pred: class label predictions + :param target: ground truth labels :param sample_weight: weight per sample for the final accuracy score. Keep None if not required. """ super().__init__(