00mjk / sklearn-onnx

Convert scikit-learn models and pipelines to ONNX

Linux	Windows

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