Multilabel Classification Python Packages

iterative-stratification

scikit-learn cross validators for iterative stratification of multilabel data

mlrl-boomer

A scikit-learn implementation of BOOMER - An Algorithm for Learning Gradient Boosted Multi-Output Rules

mlrl-common

Provides common modules to be used by different types of multi-label rule learning algorithms

mlrl-seco

A scikit-learn implementation of a Separate-and-Conquer (SeCo) multi-label rule learning algorithm

annif

Annif is a multi-algorithm automated subject indexing tool for libraries, archives and museums.

pyss3

A Python library for Interpretable Machine Learning in Text Classification using the SS3 model, with easy-to-use visualization tools for Explainable AI :octocat:

libmultilabel

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

mlrl-util

A scikit-learn implementation of BOOMER - An Algorithm for Learning Gradient Boosted Multi-Output Rules

mlrl-testbed

A scikit-learn implementation of BOOMER - An Algorithm for Learning Gradient Boosted Multi-Output Rules

torchblocks-chen

A PyTorch-based toolkit for natural language processing

mlrl-testbed-arff

A scikit-learn implementation of BOOMER - An Algorithm for Learning Gradient Boosted Multi-Output Rules

mlrl-testbed-sklearn

Adds support for the scikit-learn framework to the package "mlrl-testbed"

mlrl-testbed-slurm

A scikit-learn implementation of BOOMER - An Algorithm for Learning Gradient Boosted Multi-Output Rules

evalne

Source code for EvalNE, a Python library for evaluating Network Embedding methods.

torchblocks

A PyTorch-based toolkit for natural language processing

semantic-f1-score

Semantic F1 Score: Fair Evaluation Under Fuzzy Class Boundaries

fastxml

FastXML Extreme Multi-label Classification Algorithm

example-wise-f1-maximizer

A scikit-learn meta-estimator for multi-label classification that aims to maximize the example-wise F1 measure

maestros

A package for splitting multilabel datasets into train and test sets, while preserving the distribution of labels and keeping samples from the same group together. Includes a report and chart for visualizing the stratification.