onnxruntime的scikit学习包装器
scikit-onnxruntime的Python项目详细描述
scikit onnxruntime包装 onnxruntime 使用scikit-learnapi。
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]))
安装
您可以从PyPi：
安装pip install skonnxrt
或者您可以从源代码安装最新更改：
pip install git+https://github.com/xadupre/scikit-onnxruntime.git
贡献
我们欢迎以反馈、想法或代码的形式作出贡献。
许可证
欢迎加入QQ群-->： 979659372
