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
onnxcustom-0.4.293.tar.gz
(68.2 kB
view hashes)
Built Distribution
Close
Hashes for onnxcustom-0.4.293-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b565d24121a49a4f6c737e0b83e14fcb46f46d2026853bb2a5ee1f8de207d915 |
|
MD5 | 43e584998fac1847dbe6ab98f4540b1e |
|
BLAKE2-256 | 0830b4ccafc2327b418d046a7a2ad8d2befecfe02a1c9fd1d045f96a74c7c49f |