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.