added documentation for metrics

BiomedSciAI · Jun 30, 2024 · 8aed388 · 8aed388
1 parent cb6607f
commit 8aed388
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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__(