gorkaartola commited on
Commit
6144c8c
1 Parent(s): 009f172

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
@@ -108,7 +108,7 @@ class metric_tp_fp_Datasets(evaluate.Metric):
108
  predictions.append(predicted_indexes)
109
  print("JAAAAAAAAAARL")
110
  print(predictions)
111
- return predictions
112
  def softmax_threshold(logits, threshold):
113
  predictions = []
114
  softmax = torch.softmax(logits, dim=-1)
 
108
  predictions.append(predicted_indexes)
109
  print("JAAAAAAAAAARL")
110
  print(predictions)
111
+ return predictions
112
  def softmax_threshold(logits, threshold):
113
  predictions = []
114
  softmax = torch.softmax(logits, dim=-1)