Confusion Matrixes are essential for evaluating classifiers, but for some who are new to them, they can cause, well, confusion.
Sankey Diagrams are an alternative way of representing matrix data, and I’ve found some people – who are new to matrix data, like business domain experts who are not experienced data scientists – find them easier to understand. Also, some machine learning researchers find Sankey diagrams useful for analysing data and classifiers.
So, I have posted simple code for visualising classifier evaluation or comparisons as Sankey diagrams. Maybe it will be useful for others, as well as fun for me.
The scenarios supported are:
- Evaluating a binary classifier against ground truth or as champion-challenger,
- Evaluating a multi-class classifier against ground truth or as champion-challenger,
- Comparing multiple versions of a binary classifier, for instance over time, or hyper-parameter sweeps, and
- Comparing multiple versions of a multi-class classifier.
See the code on Github.