Spaces:
Runtime error
Runtime error
Commit
•
f5f05cf
1
Parent(s):
874887e
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 = []):
|
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'))
|