Spaces:
Runtime error
Runtime error
gorkaartola
commited on
Commit
•
6144c8c
1
Parent(s):
009f172
Update metric_for_tp_fp_samples.py
Browse files
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 |
-
|
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)
|