scikit-onnxruntime wraps onnxruntime with scikit-learn API.

with open("rf_iris.onnx", "rb") as f:
    content = f.read()

ot = OnnxTransformer(content, output_name="output_probability")
ot.fit(X_train, y_train)

print(ot.transform(X_test[:5]))

Full documentation including tutorials is available at xadupre.github.io.

You may also find answers in existing issues or submit a new one.

You can install from PyPi:

pip install skonnxrt

Or you can install from the source with the latest changes:

pip install git+https://github.com/xadupre/scikit-onnxruntime.git

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

MIT License