将 SageMaker XGBoost 模型转换为 ONNX 模型时出错

Question

我正在尝试将 SageMaker XGBoost 模型转换为 ONNX，以便在使用 ML.NET 的 .Net 应用程序中使用 ONNX 模型。我尝试使用winmltools和转换模型，onnxmltools但两个工具都返回了类似的错误。

有一个很好的资源可以在业务领域使用机器学习。我尝试在 SageMaker 中使用机器学习来提高销售额来创建模型，然后将模型转换为 ONNX 模型。该示例在 SageMaker 中运行良好。

运行示例后，我得到了一个模型，模型的类型是sagemaker.estimator.Estimator. 我试图通过使用winmltools和转换模型onnxmltools。但两者都返回相同的错误。

ValueError: No proper operator name found for '<class 'sagemaker.estimator.Estimator'>'

我尝试使用 WinMLTools遵循将 ML 模型转换为 ONNX，并且ONNXMLTools 可以将模型转换为 ONNX，从而将 SageMaker 模型转换为 ONNX 模型。

之后，我使用xgb.create_model()命令创建了 SageMaker 模型。然后使用工具将模型转换为 ONNX。但没有运气。这次我遇到了同样的错误。只是型号不一样。

ValueError: No proper operator name found for '<class 'sagemaker.model.Model'>'

然后我使用加载模型pickle并尝试转换模型。我有同样的错误，只是模型不同。

ValueError: No proper operator name found for '<class 'xgboost.core.Booster'>'

此刻，我对这些问题一无所知。我应该如何解决这些问题。我已将改进销售分类附加到 ONNX 笔记本文件以供参考。你能看看这些问题，让我知道解决问题的方法吗？提前致谢！