xgboost - 无法使用 ONNX 运行时“set_base_margin”和“使用模型的 best_ntree_limit 预测”对 XGBoost 模型进行预测

Question

我已经使用 onnxmltools 将 XGBoost 模型转换为 ONNX 模型。但是，当我尝试使用 ONNX 运行时执行预测时，我遇到了两个问题。

1. 在 XGBoost API 中，DMatrix 有一个“set_base_margin”函数，它使用“base_margin”值初始化输出。在 ONNX 运行时，我找不到可以实现相同功能的函数。ONNX 运行时确实允许传入 NamedONNXValue 实例列表以接收预测输出，但该列表中的所有预填充值都将被忽略。
2. 在 XGBoost API 中，“predict”函数有一个“ntree_limit”参数，该参数在 ONNX 运行时中不存在。

任何帮助将不胜感激。