/scikit-learn-extra

scikit-learn contrib estimators

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

PyPi Azure Codecov CircleCI ReadTheDocs

scikit-learn-extra - A set of useful tools compatible with scikit-learn

scikit-learn-extra is a Python module for machine learning that extends scikit-learn. It includes algorithms that are useful but do not satisfy the scikit-learn inclusion criteria, for instance due to their novelty or lower citation number.

Installation

Dependencies

scikit-learn-extra requires,

  • Python (>=3.6)
  • scikit-learn (>=0.22), and its dependencies

User installation

Latest release can be installed with conda,

conda install -c conda-forge scikit-learn-extra

or from PyPi with,

pip install scikit-learn-extra

Note that installing from PyPi requires a working C compiler (cf installation instructions).

The developement version can be installed with,

pip install https://github.com/scikit-learn-contrib/scikit-learn-extra/archive/master.zip

License

This package is released under the 3-Clause BSD license.