module onnx_conv.operator_converters.conv_xgboost#

Inheritance diagram of mlprodict.onnx_conv.operator_converters.conv_xgboost

Short summary#

module mlprodict.onnx_conv.operator_converters.conv_xgboost

Modified converter from XGBoost.py.

source on GitHub

Classes#

class

truncated documentation

XGBClassifierConverter

converter for XGBClassifier

XGBConverter

common methods for converters

XGBRegressorConverter

converter class

Functions#

function

truncated documentation

convert_xgboost

This converters reuses the code from XGBoost.py

Static Methods#

staticmethod

truncated documentation

_add_node

_add_node

_add_node

_fill_node_attributes

_fill_node_attributes

_fill_node_attributes

_get_default_tree_attribute_pairs

_get_default_tree_attribute_pairs

_get_default_tree_attribute_pairs

_remap_nodeid

_remap_nodeid

_remap_nodeid

common_members

common to regresssor and classifier

common_members

common to regresssor and classifier

common_members

common to regresssor and classifier

convert

convert method

convert

converter method

fill_tree_attributes

fills tree attributes

fill_tree_attributes

fills tree attributes

fill_tree_attributes

fills tree attributes

get_xgb_params

Retrieves parameters of a model.

get_xgb_params

Retrieves parameters of a model.

get_xgb_params

Retrieves parameters of a model.

validate

validate

validates the model

validate

Documentation#

Modified converter from XGBoost.py.

source on GitHub

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.

source on GitHub

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.

source on GitHub