Skip to main content

Extends scikit-learn with a couple of new models, transformers, metrics, plotting.

Project description

https://github.com/sdpython/mlinsights/blob/master/_doc/sphinxdoc/source/phdoc_static/project_ico.png?raw=true

mlinsights - extensions to scikit-learn

Build status Build Status Windows https://circleci.com/gh/sdpython/mlinsights/tree/master.svg?style=svg https://dev.azure.com/xavierdupre3/mlinsights/_apis/build/status/sdpython.mlinsights%20(2) https://badge.fury.io/py/mlinsights.svg MIT License https://codecov.io/github/sdpython/mlinsights/coverage.svg?branch=master GitHub Issues Notebook Coverage Downloads Forks Stars size

mlinsights extends scikit-learn with a couple of new models, transformers, metrics, plotting. It provides new trainers such as QuantileLinearRegression which trains a linear regression with L1 norm non-linear correlation based on decision trees, or QuantileMLPRegressor a modification of scikit-learn’s MLPRegressor which trains a multi-layer perceptron with L1 norm. It also explores PredictableTSNE which trains a supervized model to replicate t-SNE results or a PiecewiseRegression which partitions the data before fitting a model on each bucket.

Function pipeline2dot converts a pipeline into a graph:

from mlinsights.plotting import pipeline2dot
dot = pipeline2dot(clf, df)
https://raw.githubusercontent.com/sdpython/mlinsights/master/_doc/sphinxdoc/source/pipeline.png

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

mlinsights-0.3.631.tar.gz (952.5 kB view hashes)

Uploaded source

Built Distributions

mlinsights-0.3.631-cp39-cp39-win_amd64.whl (490.3 kB view hashes)

Uploaded cp39

mlinsights-0.3.631-cp38-cp38-win_amd64.whl (489.5 kB view hashes)

Uploaded cp38

mlinsights-0.3.631-cp37-cp37m-win_amd64.whl (484.9 kB view hashes)

Uploaded cp37

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page