无法使用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中，“预测”函数具有一个“ ntree_limit”参数，该参数在ONNX运行时中不存在。

任何帮助将不胜感激。