Skip to main content

Extends the list of supported operators in onnx reference implementation and onnxruntime, or implements faster versions in C++.

Project description

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

mlinsights: extensions to scikit-learn

https://dev.azure.com/xavierdupre3/mlinsights/_apis/build/status%2Fsdpython.mlinsights%20(2)?branchName=main https://badge.fury.io/py/mlinsights.svg MIT License https://codecov.io/github/sdpython/mlinsights/coverage.svg?branch=main GitHub Issues 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. PiecewiseTreeRegressor trains a piecewise regressor using a linear regression on each piece. IntervalRegressor produces confidence interval by using bootstrapping. ApproximateNMFPredictor approximates a NMF to produce prediction without retraining.

mlinsights documentation

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.5.0.tar.gz (1.7 MB view hashes)

Uploaded source

Built Distributions

mlinsights-0.5.0-cp311-cp311-win_amd64.whl (1.5 MB view hashes)

Uploaded cp311

mlinsights-0.5.0-cp310-cp310-win_amd64.whl (581.7 kB view hashes)

Uploaded cp310

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page