Spaces:
Runtime error
Runtime error
Commit
•
a325fea
1
Parent(s):
df896d1
Update metric_for_tp_fp_samples.py
Browse files
metric_for_tp_fp_samples.py
CHANGED
|
@@ -98,6 +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):
|
|
|
|
| 98 |
results = self.threshold(logits, prediction_strategy[1])
|
| 99 |
elif prediction_strategy[0] == "topk":
|
| 100 |
results = self.topk(logits, prediction_strategy[1])
|
| 101 |
+
print(results)
|
| 102 |
return results
|
| 103 |
#Prediction strategy functions______________________________________________
|
| 104 |
def argmax_max(self, logits):
|