sklearn-onnx

by onnx

Convert scikit-learn models and pipelines to ONNX

167 Stars 45 Forks Last release: about 2 months ago (1.7.1) MIT License 333 Commits 8 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

| Linux | Windows | |-------|---------| | Build Status | Build Status|

Introduction

sklearn-onnx converts scikit-learn models to ONNX. Once in the ONNX format, you can use tools like ONNX Runtime for high performance scoring.

Documentation

Full documentation including tutorials is available at http://onnx.ai/sklearn-onnx/.

You may also find answers in existing issues or submit a new one.

Installation

You can install from PyPi:

pip install skl2onnx
Or you can install from the source with the latest changes.
pip install git+https://github.com/onnx/sklearn-onnx.git

If you install sklearn-onnx from its source code, you must set the environment variable

ONNX_ML=1
before installing the onnx package.

Contribute

We welcome contributions in the form of feedback, ideas, or code.

License

MIT License

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.