sklearn-onnx:将Scikit-Learn模型/pipeline转换成ONNX格式

sklearn-onnx:将Scikit-Learn模型/pipeline转换成ONNX格式

Python 机器学习

访问GitHub主页

共165Star

详细介绍

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

推荐源码