mlrl-seco 0.15.0

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

Multi-label Separate-and-Conquer Rule Learning Algorithm

🔗 Important links: Documentation | Issue Tracker | Changelog | License

This software package provides an implementation of a Multi-label Separate-and-Conquer (SeCo) Rule Learning Algorithm that integrates with the popular scikit-learn machine learning framework.

The goal of multi-label classification is the automatic assignment of sets of labels to individual data points, for example, the annotation of text documents with topics. The algorithm that is provided by this package uses the SeCo paradigm for learning interpretable rule lists.

🔧 Functionalities

The algorithm that is provided by this project currently supports the following core functionalities to learn a binary classification rules:

• A large variety of heuristics is available to assess the quality of candidate rules.
• Rules may predict for a single label or multiple ones (which enables to model local label dependencies).
• Rules can be constructed via a greedy search or a beam search. The latter may help to improve the quality of individual rules.
• Sampling techniques and stratification methods can be used to learn new rules on a subset of the available training examples, features, or labels.
• Fine-grained control over the specificity/generality of rules is provided via hyperparameters.
• Incremental reduced error pruning can be used to remove overly specific conditions from rules and prevent overfitting.
• Sequential post-optimization may help to improve the predictive performance of a model by reconstructing each rule in the context of the other rules.
• Native support for numerical, ordinal, and nominal features eliminates the need for pre-processing techniques such as one-hot encoding.
• Handling of missing feature values, i.e., occurrences of NaN in the feature matrix, is implemented by the algorithm.

⌚ Runtime and Memory Optimizations

In addition, the following features that may speed up training or reduce the memory footprint are currently implemented:

• Sparse feature matrices can be used for training and prediction. This may speed up training significantly on some datasets.
• Sparse label matrices can be used for training. This may reduce the memory footprint in case of large datasets.
• Sparse prediction matrices can be used to store predicted labels. This may reduce the memory footprint in case of large datasets.
• Multi-threading can be used to parallelize the evaluation of a rule's potential refinements across several features or to obtain predictions for several examples in parallel.

📚 Documentation

Our documentation provides an extensive user guide, as well as Python and C++ API references for developers. If you are new to the project, you probably want to read about the following topics:

• Instructions for installing the software package or building the project from source.
• Examples of how to use the algorithm in your own Python code or how to use the command line API.
• An overview of available parameters.

A collection of benchmark datasets that are compatible with the algorithm are provided in a separate repository.

For an overview of changes and new features that have been included in past releases, please refer to the changelog.

📜 License

This project is open source software licensed under the terms of the MIT license. We welcome contributions to the project to enhance its functionality and make it more accessible to a broader audience. A frequently updated list of contributors is available here.

All contributions to the project and discussions on the issue tracker are expected to follow the code of conduct.