sklearn-onnx:将Scikit-Learn模型/pipeline转换成ONNX格式
sklearn-onnx:将Scikit-Learn模型/pipeline转换成ONNX格式
Python 机器学习
共165Star
详细介绍
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.