We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5935b56 commit 9e5723bCopy full SHA for 9e5723b
onnxmltools/convert/xgboost/operator_converters/XGBoost.py
@@ -19,6 +19,7 @@ class XGBConverter:
19
This class provides methods to validate the model, retrieve parameters,
20
and fill in the attributes for the ONNX TreeEnsemble node.
21
"""
22
+
23
@staticmethod
24
def get_xgb_params(xgb_node):
25
@@ -234,6 +235,7 @@ class XGBRegressorConverter(XGBConverter):
234
235
It handles the conversion of model parameters, tree structure,
236
and the creation of the ONNX TreeEnsembleRegressor node.
237
238
239
240
def validate(xgb_node):
241
return XGBConverter.validate(xgb_node)
0 commit comments