Spaces:
Running
Running
Update fbeta_score.py
Browse files- fbeta_score.py +1 -1
fbeta_score.py
CHANGED
@@ -95,7 +95,7 @@ class F_Beta(evaluate.Metric):
|
|
95 |
|
96 |
def _compute(self, predictions, references, beta=0.5, labels=None, pos_label=1, average="binary", sample_weight=None):
|
97 |
score = fbeta_score(
|
98 |
-
references, predictions, beta, labels=labels, pos_label=pos_label, average=average, sample_weight=sample_weight
|
99 |
)
|
100 |
return {"f_beta_score": float(score) if score.size == 1 else score}
|
101 |
|
|
|
95 |
|
96 |
def _compute(self, predictions, references, beta=0.5, labels=None, pos_label=1, average="binary", sample_weight=None):
|
97 |
score = fbeta_score(
|
98 |
+
references, predictions, beta=beta, labels=labels, pos_label=pos_label, average=average, sample_weight=sample_weight
|
99 |
)
|
100 |
return {"f_beta_score": float(score) if score.size == 1 else score}
|
101 |
|