/sklearn-onnx

Convert scikit-learn models and pipelines to ONNX

Primary LanguagePythonApache License 2.0Apache-2.0

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. All converters are tested with onnxruntime.

Documentation

Full documentation including tutorials is available at https://onnx.ai/sklearn-onnx/. Supported scikit-learn Models Last supported opset is 15.

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

Contribute

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

License

Apache License v2.0