I recently received the following question via email:
Hi Jason, quick question. A case of class imbalance: 90 cases of thumbs up 10 cases of thumbs down. How would we calculate random guessing accuracy in this case?
We can answer this question using some basic probability (I opened excel and typed in some numbers).
Let’s say the split is 90%-10% for class 0 and class 1. Let’s also say that you will guess randomly using the same ratio.
The theoretical accuracy of random guessing on a two-classification problem is:
= P(class is 0) * P(you guess 0) + P(class is 1) * P(you guess 1)
We can test this on our example 90%-10% split:
= (0.9 * 0.9) + (0.1 * 0.1)
= 0.82 * 100 or 82%
To check the math, you can plug-in a 50%-50% split of your data and it matches your intuition:
= (0.5 * 0.5) + (0.5 * 0.5)
= 0.5 * 100 or 50%
If we look on Google, we find a similar question on Cross Validated “What is the chance level accuracy in unbalanced classification problems?” with an almost identical answer. Again, a nice confirmation.
Interesting, but there is an important takeaway point from all of this.
Don’t Use Random Guessing As A Baseline
If you are looking for a classifier to use as a baseline accuracy, don’t use random guessing.
There is a classifier called Zero Rule (or 0R or ZeroR for short). It is the simplest rule you can use on a classification problem and it simply predicts the majority class in your dataset (e.g. the mode).
In the example above with a 90%-10% for class 0 and class 1 it would predict class 0 for every prediction and achieve an accuracy of 90%. This is 8% better than the theoretical maximum using random guessing.
Use the Zero Rule method as a baseline.
Also, in imbalanced classification problems like this, you should use metrics other than Accuracy such as Kappa or Area under ROC Curve.
For more information about alternative performance measures on classification problems see the post:
For more on working with imbalanced classification problems see the post:
Do you have any questions about this post? Ask in the comments.
Frustrated With Machine Learning Math?
See How Algorithms Work in Minutes
…with just arithmetic and simple examples
Discover how in my new Ebook: Master Machine Learning Algorithms
It covers explanations and examples of 10 top algorithms, like:
Linear Regression, k-Nearest Neighbors, Support Vector Machines and much more…
Finally, Pull Back the Curtain on
Machine Learning Algorithms
Skip the Academics. Just Results.