Commit 035cf28f authored by vsag96's avatar vsag96
Browse files

Update balance_acuracy score



Fixes #1891
Signed-off-by: default avatarvsag96 <tv400@snu.edu.in>
parent 6053cf9d
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -109,7 +109,8 @@ def balanced_accuracy_score(y, y_pred):
  pos_weight = num_negative / num_positive
  weights = np.ones_like(y)
  weights[y != 0] = pos_weight
  return accuracy_score(y, y_pred, sample_weight=weights)
  return sklearn.metrics.balanced_accuracy_score(
      y, y_pred, sample_weight=weights)


def pearson_r2_score(y, y_pred):