Multiclass Classification Python Packages

pycm

Multi-class confusion matrix library in Python

lightautoml

Fast and customizable framework for automatic ML model creation (AutoML)

treeig

Exact Integrated Gradients for tree ensembles, including XGBoost, LightGBM, Random Forests, and Gradient Boosting. Computes fast, exact feature attributions for tree models without numerical gradients, numerical integration, Monte Carlo sampling, or path approximations.

edef

Feature attribution for realized model fit (rather than predictions). EDEF provides exact additive decomposition of predictive performance for linear, PyTorch, JAX, and tree-based models, including standard errors, grouped inference, and exact TreeIG decompositions for tree-based models.

multiclean

Python library for morphological cleaning of multiclass 2D numpy arrays (edge smoothing and island removal

libmultilabel

This library has been moved to https://github.com/ntumlgroup/LibMultiLabel

gensvm

Python package for the GenSVM classifier

m3gp

An easy-to-use scikit-learn inspired implementation of the Multidimensional Multiclass Genetic Programming with Multidimensional Populations (M3GP) algorithm

imlightgbm

LightGBM for label-imbalanced data with focal and weighted loss function

vaganboostktf

Hybrid VAE-GAN with LightGBM for class-imbalanced classification

vaganboost

A hybrid model combining VAE, GAN, and LightGBM for boosting performance in high-energy physics or data analysis tasks.