Scikit-learn的onnxruntime包装器
项目描述
scikit-onnxruntime 使用 onnxruntime 和 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]))
文档
完整文档（包括教程）可在 xadupre.github.io 查找。
您也可以在 现有问题 中找到答案或提交新问题。
安装
您可以从 PyPi 安装。
pip install skonnxrt
或者您也可以使用最新更改从源安装
pip install git+https://github.com/xadupre/scikit-onnxruntime.git
贡献
我们欢迎以反馈、想法或代码的形式进行贡献。
许可证
关闭
哈希值 for scikit_onnxruntime-0.2.1.4-py2.py3-none-any.whl
| 算法 | 哈希摘要 | |
|---|---|---|
| SHA256 | ecbd79523a027851a6822bdd9a21dd948eab6a3d20009981dbb6cd3ec35a8496 |
|
| MD5 | f457917475b7e8239079119724034f08 |
|
| BLAKE2b-256 | 358fc5d7c687b73b1f32522b07c56324a00b5bdc25cb762088295b5936d1b0e4 |