English

The Effect of Class Imbalance on Precision-Recall Curves

Machine Learning 2021-04-28 v3 Machine Learning

Abstract

In this note I study how the precision of a classifier depends on the ratio rr of positive to negative cases in the test set, as well as the classifier's true and false positive rates. This relationship allows prediction of how the precision-recall curve will change with rr, which seems not to be well known. It also allows prediction of how FβF_{\beta} and the Precision Gain and Recall Gain measures of Flach and Kull (2015) vary with rr.

Keywords

Cite

@article{arxiv.2007.01905,
  title  = {The Effect of Class Imbalance on Precision-Recall Curves},
  author = {Christopher K I Williams},
  journal= {arXiv preprint arXiv:2007.01905},
  year   = {2021}
}

Comments

4 pages, 1 figure. Added ref to Siblini et al (2020) and last sentence. Final m/s version of paper published in Neural Computation