Microsoft ML Tool to convert machine learning models to ONNX for use with Windows Machine Learning
Project description
Introduction
WinMLTools enables you to convert models from different machine learning toolkits into ONNX for use with Windows Machine Learning. Currently the following toolkits are supported:
Apple CoreML
scikit-learn (subset of models convertible to ONNX)
LibSVM
XGBoost
Install
pip install winmltools
Dependancies
This converter package extends the functionalities of ONNXMLTools.
scikit-learn is needed to convert a scikit-learn model, coremltools for Apple CoreML.
Example
Here is a simple example to convert a CoreML model:
import winmltools
import coremltools
model_coreml = coremltools.utils.load_spec("image_recognition.mlmodel")
model_onnx = winmltools.convert.convert_coreml(model_coreml, "Image_Reco")
# Save as text
winmltools.utils.save_text(model_onnx, "image_recognition.json")
# Save as protobuf
winmltools.utils.save_model(model_onnx, "image_recognition.onnx")
License
MIT License
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
File details
Details for the file winmltools-0.1.0.5072-py2.py3-none-any.whl.
File metadata
- Download URL: winmltools-0.1.0.5072-py2.py3-none-any.whl
- Upload date:
- Size: 20.1 kB
- Tags: Python 2, Python 3
- Uploaded using Trusted Publishing? No
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | c7960d9b59bffc46c9f744cf3933b4663d8d811ed87e888eb0d6bf959fea82bc |
|
| MD5 | a982e61ee92f78ffa29bdb66a91faeed |
|
| BLAKE2b-256 | b972696567cb073132ada1dec96fb615f8c8117ab6114dd18f010ccd2529d325 |
