• Stars
    star
    548
  • Rank 81,119 (Top 2 %)
  • Language
    Python
  • License
    Apache License 2.0
  • Created almost 6 years ago
  • Updated about 2 months ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Convert scikit-learn models and pipelines to ONNX

Linux Windows
Build Status Build Status

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.

Documentation

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

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

Contribute

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

License

Apache License v2.0

More Repositories

1

onnx

Open standard for machine learning interoperability
Python
17,846
star
2

models

A collection of pre-trained, state-of-the-art models in the ONNX format
Jupyter Notebook
7,054
star
3

tutorials

Tutorials for creating and using ONNX models
Jupyter Notebook
3,175
star
4

onnx-tensorrt

ONNX-TensorRT: TensorRT backend for ONNX
C++
2,922
star
5

tensorflow-onnx

Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX
Jupyter Notebook
2,303
star
6

onnx-tensorflow

Tensorflow Backend for ONNX
Python
1,281
star
7

onnxmltools

ONNXMLTools enables conversion of models to ONNX
Python
1,002
star
8

onnx-mlir

Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
C++
754
star
9

optimizer

Actively maintained ONNX Optimizer
C++
579
star
10

onnx-coreml

ONNX to Core ML Converter
Python
393
star
11

keras-onnx

Convert tf.keras/Keras models to ONNX
Python
379
star
12

onnx-caffe2

Caffe2 implementation of Open Neural Network Exchange (ONNX)
Python
165
star
13

onnx-docker

Dockerfiles and scripts for ONNX container images
Jupyter Notebook
133
star
14

onnx-mxnet

ONNX model format support for Apache MXNet
Python
96
star
15

turnkeyml

The AI insights toolchain
Python
53
star
16

onnx-r

R Interface to Open Neural Network Exchange (ONNX)
R
44
star
17

backend-scoreboard

Scoreboard for ONNX Backend Compatibility
Python
24
star
18

onnx.github.io

Code of the official webpage of onnx
HTML
22
star
19

steering-committee

Notes and artifacts from the ONNX steering committee
Jupyter Notebook
22
star
20

working-groups

Repository for ONNX working group artifacts
Jupyter Notebook
20
star
21

sigs

Repository for ONNX SIG artifacts
19
star
22

onnx-xla

XLA integration of Open Neural Network Exchange (ONNX)
C++
19
star
23

wheel-builder

Utils for building and publishing ONNX wheels
Shell
7
star
24

onnx-cntk

Python
6
star