如何解决无法使用ONNX运行时“ set_base_margin”和“使用模型的best_ntree_limit进行预测”来对XGBoost模型进行预测
我已使用onnxmltools将XGBoost模型转换为ONNX模型。 但是,当我尝试使用ONNX运行时执行预测时,遇到两个问题。
- 在XGBoost API中,DMatrix具有一个“ set_base_margin”函数,该函数使用“ base_margin”值初始化输出。在ONNX运行时中,找不到能够实现相同功能的功能。 ONNX运行时确实允许传入NamedONNXValue实例的列表以接收预测输出,但是该列表中的所有预填充值都会被忽略。
- 在XGBoost API中,“预测”函数具有一个“ ntree_limit”参数,该参数在ONNX运行时中不存在。
任何帮助将不胜感激。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。