Skip to main content

Convert scikit-learn models to ONNX

Project description

Linux

Windows/Macos

Code style: black

Introduction

sklearn-onnx converts scikit-learn models to ONNX. Once in the ONNX format, you can use tools like ONNX Runtime for high performance scoring. All converters are tested with onnxruntime. Any external converter can be registered to convert scikit-learn pipeline including models or transformers coming from external libraries.

Documentation

Full documentation including tutorials is available at https://onnx.ai/sklearn-onnx/. Supported scikit-learn Models Last supported opset is 21.

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

Installation

You can install from PyPi:

pip install skl2onnx

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

pip install git+https://github.com/onnx/sklearn-onnx.git

Getting started

# Train a model.
import numpy as np
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.ensemble import RandomForestClassifier

iris = load_iris()
X, y = iris.data, iris.target
X = X.astype(np.float32)
X_train, X_test, y_train, y_test = train_test_split(X, y)
clr = RandomForestClassifier()
clr.fit(X_train, y_train)

# Convert into ONNX format.
from skl2onnx import to_onnx

onx = to_onnx(clr, X[:1])
with open("rf_iris.onnx", "wb") as f:
    f.write(onx.SerializeToString())

# Compute the prediction with onnxruntime.
import onnxruntime as rt

sess = rt.InferenceSession("rf_iris.onnx", providers=["CPUExecutionProvider"])
input_name = sess.get_inputs()[0].name
label_name = sess.get_outputs()[0].name
pred_onx = sess.run([label_name], {input_name: X_test.astype(np.float32)})[0]

Contribute

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

License

Apache License v2.0

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

skl2onnx-1.19.1.tar.gz (948.9 kB view details)

Uploaded Source

Built Distribution

skl2onnx-1.19.1-py3-none-any.whl (315.5 kB view details)

Uploaded Python 3

File details

Details for the file skl2onnx-1.19.1.tar.gz.

File metadata

  • Download URL: skl2onnx-1.19.1.tar.gz
  • Upload date:
  • Size: 948.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.12.9

File hashes

Hashes for skl2onnx-1.19.1.tar.gz
Algorithm Hash digest
SHA256 0c105f2a3b87a624dd218d1fb98fdd19cf1bf6217190d25ce7e15484127d0e5d
MD5 331c2f01380f2bce278782568a933f12
BLAKE2b-256 e1593f4d82eb96d8a80fd80325fc658d20e772ee6ac1c6f5cc85dd6aaaf7ccff

See more details on using hashes here.

Provenance

The following attestation bundles were made for skl2onnx-1.19.1.tar.gz:

Publisher: release.yml on onnx/sklearn-onnx

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file skl2onnx-1.19.1-py3-none-any.whl.

File metadata

  • Download URL: skl2onnx-1.19.1-py3-none-any.whl
  • Upload date:
  • Size: 315.5 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.12.9

File hashes

Hashes for skl2onnx-1.19.1-py3-none-any.whl
Algorithm Hash digest
SHA256 fddf2f49e3ffc355f332e676b43c6fec5e63797627925b279d9f5b2c4d0c81a7
MD5 dfd428645ed29b9e76a99abdbf63a89a
BLAKE2b-256 57ec9a0d709217aa385d87b3eadcf19e2ae32eca097077fa2236312d5fc8f656

See more details on using hashes here.

Provenance

The following attestation bundles were made for skl2onnx-1.19.1-py3-none-any.whl:

Publisher: release.yml on onnx/sklearn-onnx

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page