Metadata-Version: 2.1
Name: hipe4ml-converter
Version: 0.0.6
Summary: Minimal heavy ion physics environment for Machine Learning
Home-page: https://github.com/hipe4ml/hipe4ml_converter
Author: hipe4ml-developers
Author-email: hipe4ml@googlegroups.com
License: GPL
Platform: UNKNOWN
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Education
Classifier: Intended Audience :: Developers
Classifier: Topic :: Scientific/Engineering :: Physics
Classifier: License :: OSI Approved :: GNU General Public License v3 (GPLv3)
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Requires-Python: >=3.8
Description-Content-Type: text/markdown
Requires-Dist: psutil
Requires-Dist: dutil
Requires-Dist: pyarrow >=5.0.0
Requires-Dist: hipe4ml >=0.0.15
Requires-Dist: ipython >=7.16.1
Requires-Dist: jedi ==0.17.2
Requires-Dist: torch >=1.7.0
Requires-Dist: scikit-learn >=0.21.3
Requires-Dist: onnxconverter-common >=1.13.0
Requires-Dist: skl2onnx >=1.12.0
Requires-Dist: onnxmltools <=1.11.0,>=1.6.0
Requires-Dist: onnxruntime >=1.12.0
Requires-Dist: lightgbm <=3.3.5,>=2.2
Requires-Dist: hummingbird-ml[extra] >=0.4.9
Provides-Extra: dev
Requires-Dist: pylint >=2.6.2 ; extra == 'dev'
Requires-Dist: flake8 >=3.8.4 ; extra == 'dev'
Requires-Dist: pytest >=6.2.2 ; extra == 'dev'
Requires-Dist: twine >=3.3.0 ; extra == 'dev'
Requires-Dist: setuptools >=53.0.0 ; extra == 'dev'
Requires-Dist: wheel >=0.36.2 ; extra == 'dev'

![](https://github.com/hipe4ml/hipe4ml_converter/actions/workflows/pythonpackage.yml/badge.svg?branch=main)

![](https://github.com/hipe4ml/hipe4ml_converter/actions/workflows/pythonpublish.yml/badge.svg)



# hipe4ml_converter

Package for conversion of machine learning models trained with hipe4ml into ONNX and tensor formats



# Introduction



hipe4ml-converter enables you to convert models trained with [hipe4ml](https://github.com/hipe4ml/hipe4ml) into [ONNX](https://onnx.ai/) format and tensor formats ([PyTorch](https://pytorch.org/), [TorchScript](https://pytorch.org/docs/stable/jit.html), [TVM](https://tvm.apache.org/) or [ONNX](https://onnx.ai/)) via the [hummingbird](https://github.com/microsoft/hummingbird) library.



# Documentation



[https://hipe4ml.github.io/hipe4ml_converter/index.html](https://hipe4ml.github.io/hipe4ml_converter/index.html)



# Requirements



- [hipe4ml](https://github.com/hipe4ml/hipe4ml)

- [onnx](https://github.com/onnx/onnx)

- [onnxruntime](https://github.com/microsoft/onnxruntime)

- [onnxmltools](https://github.com/onnx/onnxmltools)

- [hummingbird](https://github.com/microsoft/hummingbird)


