Hierarchical confusion matrix for classification performance evaluation

Kevin Riehl, Michael Neunteufel, Martin Hemberg
6/15/2023
cs.LG

No official code found. Best implementation: DerKevinRiehl/HierarchicalConfusionMatrix (3 stars, last commit 2023-07-20).

Abstract

In this work we propose a novel concept of a hierarchical confusion matrix, opening the door for popular confusion matrix based (flat) evaluation measures from binary classification problems, while considering the peculiarities of hierarchical classification problems. We develop the concept to a generalized form and prove its applicability to all types of hierarchical classification problems including directed acyclic graphs, multi path labelling, and non mandatory leaf node prediction. Finally, we use measures based on the novel confusion matrix to evaluate models within a benchmark for three real world hierarchical classification applications and compare the results to established evaluation measures. The results outline the reasonability of this approach and its usefulness to evaluate hierarchical classification problems. The implementation of hierarchical confusion matrix is available on GitHub.

View on arXivView PDF

Code Implementations(1)

31Python, Jupyter NotebookApr 21, 20213 years ago

Similar Papers

Cite this paper

@article{riehl2023hierarchical,
  title  = {Hierarchical confusion matrix for classification performance evaluation},
  author = {Kevin Riehl and Michael Neunteufel and Martin Hemberg},
  year   = {2023},
  eprint = {2306.09461},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2306.09461v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2306.09461.svg)](https://codeofpaper.com/paper/2306.09461)

Discussion