Spaces:
Sleeping
Sleeping
Update fbeta_score.py
Browse files- fbeta_score.py +1 -1
fbeta_score.py
CHANGED
@@ -97,6 +97,6 @@ class F_Beta(evaluate.Metric):
|
|
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":
|
101 |
|
102 |
|
|
|
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": score if getattr(score, "size", 1) > 1 else float(score)}
|
101 |
|
102 |
|