Scikit-learn wrapper of onnxruntime
Project description
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]))
Documentation
Full documentation including tutorials is available at xadupre.github.io.
You may also find answers in existing issues or submit a new one.
Installation
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
Contribute
We welcome contributions in the form of feedback, ideas, or code.
License
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
File details
Details for the file scikit_onnxruntime-0.2.1.3-py2.py3-none-any.whl.
File metadata
- Download URL: scikit_onnxruntime-0.2.1.3-py2.py3-none-any.whl
- Upload date:
- Size: 4.9 kB
- Tags: Python 2, Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: Python-urllib/3.7
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 3985456b5995eb099bce49edfb707f418a860fad0205d17b822229ecc6def9f1 |
|
| MD5 | a304eabe3f5a09ec9f3d44b0aec4a998 |
|
| BLAKE2b-256 | 54614f0e2e4f206078e537684b248c4b8997f9fb71b2e66e6338f3629e283359 |
