jsalvad0r commited on
Commit
c153b4f
·
1 Parent(s): 536d81c
Files changed (1) hide show
  1. semf1.py +3 -0
semf1.py CHANGED
@@ -339,6 +339,7 @@ class SemF1(evaluate.Metric):
339
  gpu: DEVICE_TYPE = False,
340
  batch_size: int = 32,
341
  verbose: bool = False,
 
342
  ) -> List[Scores]:
343
  """
344
  Compute precision, recall, and F1 scores for given predictions and references.
@@ -358,6 +359,7 @@ class SemF1(evaluate.Metric):
358
  :param gpu: GPU device to use.
359
  :param batch_size: Batch size for encoding.
360
  :param verbose: Flag to indicate verbose output.
 
361
 
362
  :return: List of Scores dataclass with precision, recall, and F1 scores.
363
  """
@@ -422,3 +424,4 @@ class SemF1(evaluate.Metric):
422
 
423
  return results
424
 
 
 
339
  gpu: DEVICE_TYPE = False,
340
  batch_size: int = 32,
341
  verbose: bool = False,
342
+ aggregate: bool = True,
343
  ) -> List[Scores]:
344
  """
345
  Compute precision, recall, and F1 scores for given predictions and references.
 
359
  :param gpu: GPU device to use.
360
  :param batch_size: Batch size for encoding.
361
  :param verbose: Flag to indicate verbose output.
362
+ :param aggregate: Flag to determine if output should be averaged
363
 
364
  :return: List of Scores dataclass with precision, recall, and F1 scores.
365
  """
 
424
 
425
  return results
426
 
427
+ # TEST