gorkaartola commited on
Commit
9819d89
1 Parent(s): 303551b

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
@@ -98,7 +98,7 @@ class metric_tp_fp_Datasets(evaluate.Metric):
98
  results = self.threshold(logits, prediction_strategy[1])
99
  elif prediction_strategy[0] == "topk":
100
  results = self.topk(logits, prediction_strategy[1])
101
- return results
102
  #Prediction strategy functions______________________________________________
103
  def argmax_max(self, logits):
104
  predictions = []
 
98
  results = self.threshold(logits, prediction_strategy[1])
99
  elif prediction_strategy[0] == "topk":
100
  results = self.topk(logits, prediction_strategy[1])
101
+ return results
102
  #Prediction strategy functions______________________________________________
103
  def argmax_max(self, logits):
104
  predictions = []