Spaces:
Runtime error
Runtime error
Commit
•
11fc8af
1
Parent(s):
dbdeb1f
Update metric_for_tp_fp_samples.py
Browse files
metric_for_tp_fp_samples.py
CHANGED
|
@@ -106,8 +106,6 @@ class metric_tp_fp_Datasets(evaluate.Metric):
|
|
| 106 |
for prediction in argmax:
|
| 107 |
predicted_indexes = [prediction.item()]
|
| 108 |
predictions.append(predicted_indexes)
|
| 109 |
-
print("JAAAAAAAAAARL")
|
| 110 |
-
print(predictions)
|
| 111 |
return predictions
|
| 112 |
def softmax_threshold(logits, threshold):
|
| 113 |
predictions = []
|
|
|
|
| 106 |
for prediction in argmax:
|
| 107 |
predicted_indexes = [prediction.item()]
|
| 108 |
predictions.append(predicted_indexes)
|
|
|
|
|
|
|
| 109 |
return predictions
|
| 110 |
def softmax_threshold(logits, threshold):
|
| 111 |
predictions = []
|