Frameworks#

Before you start, you should take a look at the table below in order to check if the framework you are using is already supported by this package.

Framework

Converter

Raw Example

Supported

TensorFlow

tf2onnx

onnxruntime

Yes

Keras

tf2onnx

onnxruntime

Yes

Scikit-learn

skl2onnx

onnx

Yes

XGBoost

Native

onnxmltools

Yes

LightGBM

Native

onnxmltools

Yes

CoreML

coremltools

onnxmltools

Yes

CatBoost

Native

onnxmltools

Yes

Spark

Native

onnxmltools

Yes

H2O

Native

onnxmltools

Yes

Backend#

It is important to note that this package relies on ONNXMLTools, if you want to add a new framework, you should check if it is supported by ONNXMLTools, and if it is not, you may contribute to their package first.