Skip to main content

Extends scikit-learn with a couple of new models, transformers, metrics, plotting.

Project description

https://circleci.com/gh/sdpython/onnxcustom/tree/master.svg?style=svg Build status Build Status Windows https://codecov.io/gh/sdpython/onnxcustom/branch/master/graph/badge.svg https://badge.fury.io/py/onnxcustom.svg GitHub Issues MIT License Downloads Forks Stars size

onnxcustom: custom ONNX

https://raw.githubusercontent.com/sdpython/onnxcustom/master/_doc/sphinxdoc/source/phdoc_static/project_ico.png

documentation

Examples, tutorial on how to convert machine learned models into ONNX, implement your own converter or runtime, or even train with ONNX / onnxruntime.

The function check or the command line python -m onnxcustom check checks the module is properly installed and returns processing time for a couple of functions or simply:

import onnxcustom
onnxcustom.check()

The documentation also introduces onnx, onnxruntime for inference and training. The tutorial related to scikit-learn has been merged into sklearn-onnx documentation. Among the tools this package implements, you may find:

  • a tool to convert NVidia Profilder logs into a dataframe,

  • a SGD optimizer similar to what scikit-learn implements but based on onnxruntime-training and able to train an CPU and GPU,

  • functions to manipulate onnx graph.

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

onnxcustom-0.4.293.tar.gz (68.2 kB view hashes)

Uploaded Source

Built Distribution

onnxcustom-0.4.293-py3-none-any.whl (77.8 kB view hashes)

Uploaded Python 3

Supported by

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