问题描述
我已使用onnxmltools将XGBoost模型转换为ONNX模型。 但是，当我尝试使用ONNX运行时执行预测时，遇到两个问题。
- 在XGBoost API中，DMatrix具有一个“ set_base_margin”函数，该函数使用“ base_margin”值初始化输出。在ONNX运行时中，找不到能够实现相同功能的功能。 ONNX运行时确实允许传入NamedONNXValue实例的列表以接收预测输出，但是该列表中的所有预填充值都会被忽略。
- 在XGBoost API中，“预测”函数具有一个“ ntree_limit”参数，该参数在ONNX运行时中不存在。
任何帮助将不胜感激。
解决方法
暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！
如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@）