Skip to content

Commit 4256523

Browse files
committed
rename balanced_acc to acc_balanced
1 parent eea4f44 commit 4256523

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/arviz_stats/loo.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -275,7 +275,7 @@ def loo_metrics(data, kind="rmse", var_name=None, round_to="2g"):
275275
- 'mse': mean squared error.
276276
- 'rmse': root mean squared error. Default.
277277
- 'acc': classification accuracy.
278-
- 'balanced_acc': balanced classification accuracy.
278+
- 'acc_balanced': balanced classification accuracy.
279279
280280
var_name: str, optional
281281
The name of the variable in log_likelihood groups storing the pointwise log
@@ -318,7 +318,7 @@ def loo_metrics(data, kind="rmse", var_name=None, round_to="2g"):
318318
Journal of Machine Learning Research, 25(72) (2024) https://jmlr.org/papers/v25/19-556.html
319319
arXiv preprint https://arxiv.org/abs/1507.02646
320320
"""
321-
valid_kind = ["mae", "rmse", "mse", "acc", "balanced_acc"]
321+
valid_kind = ["mae", "rmse", "mse", "acc", "acc_balanced"]
322322
if kind not in valid_kind:
323323
raise ValueError(f"kind must be one of {valid_kind}")
324324

0 commit comments

Comments
 (0)