Преобразование в формат ONNX (ONNXMLTools)
ONNXMLTools позволяет преобразовывать модели из различных наборов средств машинного обучения в формат ONNX.
Инструкции по установке и использованию доступны в репозитории ONNXMLTools на GitHub.
Поддержка
В настоящее время поддерживаются следующие наборы средств:
- Keras (оболочка преобразователя keras2onnx);
- Tensorflow (оболочка преобразователя tf2onnx);
- scikit-learn (оболочка преобразователя skl2onnx);
- Apple Core ML;
- Spark ML (экспериментальный режим);
- LightGBM
- libscm;
- XGBoost
- H2O
- CatBoost
Pytorch также имеет встроенное средство экспорта ONNX. Дополнительные сведения см. здесь.
Примечание.
Используйте следующие ресурсы для получения справки по машинному обучению в Windows:
- Чтобы задать технические вопросы о машинном обучении в Windows или ответить на них, используйте тег windows-machine-learning в Stack Overflow.
- Сообщить об ошибке можно в нашем репозитории GitHub.