[go: up one dir, main page]

Skip to content

Convert scikit-learn models and pipelines to ONNX

License

Notifications You must be signed in to change notification settings

szha/sklearn-onnx

 
 

Repository files navigation

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

About

Convert scikit-learn models and pipelines to ONNX

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%