Spaces:
Runtime error
Runtime error
gorkaartola
commited on
Commit
•
4b97a39
1
Parent(s):
6b7c88e
Update metric_for_tp_fp_samples.py
Browse files
metric_for_tp_fp_samples.py
CHANGED
@@ -191,7 +191,7 @@ class metric_tp_fp_Datasets(evaluate.Metric):
|
|
191 |
return results
|
192 |
|
193 |
#Computes the metric for each prediction strategy##############################################
|
194 |
-
def _compute(self, predictions, references, prediction_strategies = ["argmax_max"]):
|
195 |
"""Returns the scores"""
|
196 |
# TODO: Compute the different scores of the metric
|
197 |
predictions = torch.from_numpy(np.array(predictions, dtype = 'float32'))
|
|
|
191 |
return results
|
192 |
|
193 |
#Computes the metric for each prediction strategy##############################################
|
194 |
+
def _compute(self, predictions, references, prediction_strategies = [["argmax_max"],]):
|
195 |
"""Returns the scores"""
|
196 |
# TODO: Compute the different scores of the metric
|
197 |
predictions = torch.from_numpy(np.array(predictions, dtype = 'float32'))
|