Преобразование ONNX (ONNXMLTools)

ONNXMLTools позволяет преобразовать модели из различных средств машинного обучения в ONNX.

Инструкции по установке и использованию доступны в репозитории ONNXMLTools GitHub.

Поддержка

В настоящее время поддерживаются следующие наборы средств.

• Keras (оболочка преобразователя keras2onnx)
• Tensorflow (оболочка конвертера tf2onnx)
• scikit-learn (оболочка преобразователя skl2onnx)
• Apple Core ML
• Spark ML (экспериментальный)
• LightGBM
• ЛИБСКМ
• XGBoost
• H2O
• КатБуст

Pytorch также имеет встроенный экспортер ONNX. Дополнительные сведения см. здесь .