about
A Python library for hierarchical classification compatible with scikit-learn (arxiv.org)
5 points by PaulHoule on Jan 4, 2023 | hide | past | pdf | discuss on HN

In plain words: HiClass is a free Python library for classifying items whose labels form a tree, letting you attach a separate classifier to each branch instead of one flat label guesser. It plugs into the popular scikit-learn toolkit and adds scoring measures built for tree-shaped labels.

Abstract · HiClass: a Python library for local hierarchical classification compatible with scikit-learn

HiClass is an open-source Python library for local hierarchical classification entirely compatible with scikit-learn. It contains implementations of the most common design patterns for hierarchical machine learning models found in the literature, that is, the local classifiers per node, per parent node and per level. Additionally, the package contains implementations of hierarchical metrics, which are more appropriate for evaluating classification performance on hierarchical data. The documentation includes installation and usage instructions, examples within tutorials and interactive notebooks, and a complete description of the API. HiClass is released under the simplified BSD license, encouraging its use in both academic and commercial environments. Source code and documentation are available at https://github.com/scikit-learn-contrib/hiclass.

Fábio M. Miranda, Niklas Köhnecke, Bernhard Y. Renard
arXiv:2112.06560 · cs.LG · submitted Dec 13, 2021 · updated Jan 3, 2023
abstract · pdf · html · 17 pages, 9 figures, 7 tables

add comment on HN