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 ML 的帮助:
- 若要提出或回答有关 Windows ML 的技术问题,请在 Stack Overflow 上使用 windows-machine-learning 标记。
- 若要报告 bug,请在 GitHub 上提交问题。