gorkaartola commited on
Commit
f5f05cf
1 Parent(s): 874887e

Update metric_for_tp_fp_samples.py

Browse files
Files changed (1) hide show
  1. metric_for_tp_fp_samples.py +1 -1
metric_for_tp_fp_samples.py CHANGED
@@ -191,7 +191,7 @@ class metric_tp_fp_Datasets(evaluate.Metric):
191
  return results
192
 
193
  #Computes the metric for each prediction strategy##############################################
194
- def _compute(self, predictions, references, prediction_strategies = []):
195
  """Returns the scores"""
196
  # TODO: Compute the different scores of the metric
197
  predictions = torch.from_numpy(np.array(predictions, dtype = 'float32'))
 
191
  return results
192
 
193
  #Computes the metric for each prediction strategy##############################################
194
+ def _compute(self, predictions, references, prediction_strategies = ['argmax_max']):
195
  """Returns the scores"""
196
  # TODO: Compute the different scores of the metric
197
  predictions = torch.from_numpy(np.array(predictions, dtype = 'float32'))