gorkaartola commited on
Commit
4b97a39
1 Parent(s): 6b7c88e

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 = ["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'))
 
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'))