buelfhood commited on
Commit
970ee16
·
verified ·
1 Parent(s): 965b544

Update fbeta_score.py

Browse files
Files changed (1) hide show
  1. 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": float(score) if score.size == 1 else 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