Spaces:
Runtime error
Runtime error
gorkaartola
commited on
Commit
•
a260b86
1
Parent(s):
7815822
Update metric_for_tp_fp_samples.py
Browse files
metric_for_tp_fp_samples.py
CHANGED
@@ -133,7 +133,7 @@ class metric_tp_fp_Datasets(evaluate.Metric):
|
|
133 |
def topk(self, logits, topk):
|
134 |
predictions = logits.topk(topk).indices.tolist()
|
135 |
print(predictions)
|
136 |
-
|
137 |
|
138 |
#Builds a report with the metrics####################################################
|
139 |
def metrics_report(self, true_positives = "", false_positives = ""):
|
|
|
133 |
def topk(self, logits, topk):
|
134 |
predictions = logits.topk(topk).indices.tolist()
|
135 |
print(predictions)
|
136 |
+
return predictions
|
137 |
|
138 |
#Builds a report with the metrics####################################################
|
139 |
def metrics_report(self, true_positives = "", false_positives = ""):
|