leslyarun commited on
Commit
5aabb05
1 Parent(s): c15140e

Update fbeta_score.py

Browse files
Files changed (1) hide show
  1. 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