gorkaartola commited on
Commit
f151b2a
1 Parent(s): a31ef3c

Update metric_for_tp_fp_samples.py

Browse files
Files changed (1) hide show
  1. metric_for_tp_fp_samples.py +1 -0
metric_for_tp_fp_samples.py CHANGED
@@ -120,6 +120,7 @@ class metric_tp_fp_Datasets(evaluate.Metric):
120
  def softmax_topk(self, logits, topk):
121
  softmax = torch.softmax(logits, dim=-1)
122
  predictions = softmax.topk(topk).indices.tolist()
 
123
  return predictions
124
  def threshold(self, logits, threshold):
125
  predictions = []
 
120
  def softmax_topk(self, logits, topk):
121
  softmax = torch.softmax(logits, dim=-1)
122
  predictions = softmax.topk(topk).indices.tolist()
123
+ print(predictions)
124
  return predictions
125
  def threshold(self, logits, threshold):
126
  predictions = []