Interactive Classification Metrics: A graphical application to build robust intuition for classification model evaluation

David H. Brown, Davide Chicco
12/22/2024
cs.LGstat.ML

Official code: davhbrown/interactive_classification_metrics (16 stars, MIT, last commit 2024-12-29).

Abstract

Machine learning continues to grow in popularity in academia, in industry, and is increasingly used in other fields. However, most of the common metrics used to evaluate even simple binary classification models have shortcomings that are neither immediately obvious nor consistently taught to practitioners. Here we present Interactive Classification Metrics (ICM), an application to visualize and explore the relationships between different evaluation metrics. The user changes the distribution statistics and explores corresponding changes across a suite of evaluation metrics. The interactive, graphical nature of this tool emphasizes the tradeoffs of each metric without the overhead of data wrangling and model training. The goals of this application are: (1) to aid practitioners in the ever-expanding machine learning field to choose the most appropriate evaluation metrics for their classification problem; (2) to promote careful attention to interpretation that is required even in the simplest scenarios like binary classification. Our application is publicly available for free under the MIT license as a Python package on PyPI at https://pypi.org/project/interactive-classification-metrics and on GitHub at https://github.com/davhbrown/interactive_classification_metrics.

View on arXivView PDF

Code Implementations(1)

MIT

Get an intuitive sense for the ROC curve and other binary classification metrics with interactive visualization.

160PythonJan 11, 20241 years agoMIT
binary-classificationmachine-learningmatthews-correlation-coefficientroc-curve

Similar Papers

Cite this paper

@article{brown2024interactive,
  title  = {Interactive Classification Metrics: A graphical application to build robust intuition for classification model evaluation},
  author = {David H. Brown and Davide Chicco},
  year   = {2024},
  eprint = {2412.17066},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2412.17066v1}
}

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/2412.17066.svg)](https://codeofpaper.com/paper/2412.17066)

Discussion