Extends scikit-learn with a couple of new models, transformers, metrics, plotting.
Project description
onnxcustom: custom ONNX
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
Built Distribution
Hashes for onnxcustom-0.4.293-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b565d24121a49a4f6c737e0b83e14fcb46f46d2026853bb2a5ee1f8de207d915 |
|
MD5 | 43e584998fac1847dbe6ab98f4540b1e |
|
BLAKE2b-256 | 0830b4ccafc2327b418d046a7a2ad8d2befecfe02a1c9fd1d045f96a74c7c49f |