module onnx_conv.operator_converters.conv_xgboost
#
Short summary#
module mlprodict.onnx_conv.operator_converters.conv_xgboost
Modified converter from XGBoost.py.
Classes#
class |
truncated documentation |
---|---|
converter for XGBClassifier |
|
common methods for converters |
|
converter class |
Functions#
function |
truncated documentation |
---|---|
This converters reuses the code from XGBoost.py … |
Static Methods#
staticmethod |
truncated documentation |
---|---|
|
|
|
|
|
|
|
|
|
|
|
|
|
common to regresssor and classifier |
common to regresssor and classifier |
|
|
common to regresssor and classifier |
convert method |
|
converter method |
|
|
fills tree attributes |
fills tree attributes |
|
|
fills tree attributes |
|
Retrieves parameters of a model. |
Retrieves parameters of a model. |
|
|
Retrieves parameters of a model. |
validates the model |
|
Documentation#
Modified converter from XGBoost.py.
- class mlprodict.onnx_conv.operator_converters.conv_xgboost.XGBClassifierConverter#
Bases:
XGBConverter
converter for XGBClassifier
- static _get_default_tree_attribute_pairs()#
- static convert(scope, operator, container)#
convert method
- static validate(xgb_node)#
validates the model
- class mlprodict.onnx_conv.operator_converters.conv_xgboost.XGBConverter#
Bases:
object
common methods for converters
- static _add_node(attr_pairs, is_classifier, tree_id, tree_weight, node_id, feature_id, mode, value, true_child_id, false_child_id, weights, weight_id_bias, missing, hitrate)#
- static _fill_node_attributes(treeid, tree_weight, jsnode, attr_pairs, is_classifier, remap)#
- static _get_default_tree_attribute_pairs(is_classifier)#
- static _remap_nodeid(jsnode, remap=None)#
- static common_members(xgb_node, inputs)#
common to regresssor and classifier
- static fill_tree_attributes(js_xgb_node, attr_pairs, tree_weights, is_classifier)#
fills tree attributes
- static get_xgb_params(xgb_node)#
Retrieves parameters of a model.
- static validate(xgb_node)#
validates the model
- class mlprodict.onnx_conv.operator_converters.conv_xgboost.XGBRegressorConverter#
Bases:
XGBConverter
converter class
- static _get_default_tree_attribute_pairs()#
- static convert(scope, operator, container)#
converter method
- static validate(xgb_node)#
validates the model
- mlprodict.onnx_conv.operator_converters.conv_xgboost.convert_xgboost(scope, operator, container)#
This converters reuses the code from XGBoost.py and makes some modifications. It implements converters for models in xgboost.