Spaces:
Runtime error
Runtime error
Commit
•
f151b2a
1
Parent(s):
a31ef3c
Update metric_for_tp_fp_samples.py
Browse files
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 = []
|