module onnx_conv.operator_converters.parse_lightgbm#

Inheritance diagram of mlprodict.onnx_conv.operator_converters.parse_lightgbm

Short summary#

module mlprodict.onnx_conv.operator_converters.parse_lightgbm

Parsers for LightGBM booster.

source on GitHub

Classes#

class

truncated documentation

MockWrappedLightGbmBoosterClassifier

Mocked lightgbm.

WrappedLightGbmBooster

A booster can be a classifier, a regressor. Trick to wrap it in a minimal function.

WrappedLightGbmBoosterClassifier

Trick to wrap a LGBMClassifier into a class.

Functions#

function

truncated documentation

converter_lightgbm_concat

Converter for operator LightGBMConcat.

lightgbm_parser

Agnostic parser for LightGBM Booster.

shape_calculator_lightgbm_concat

Shape calculator for operator LightGBMConcat.

Static Methods#

staticmethod

truncated documentation

_generate_classes

Methods#

method

truncated documentation

__init__

__init__

__init__

attr

Returns default values for common attributes.

dump_model

mock dump_model method

feature_name

Returns binary features names.

get_objective

Returns the objective.

Documentation#

Parsers for LightGBM booster.

source on GitHub

class mlprodict.onnx_conv.operator_converters.parse_lightgbm.MockWrappedLightGbmBoosterClassifier(tree)#

Bases: WrappedLightGbmBoosterClassifier

Mocked lightgbm.

source on GitHub

__init__(tree)#
attr(key)#

Returns default values for common attributes.

dump_model()#

mock dump_model method

feature_name()#

Returns binary features names.

class mlprodict.onnx_conv.operator_converters.parse_lightgbm.WrappedLightGbmBooster(booster)#

Bases: object

A booster can be a classifier, a regressor. Trick to wrap it in a minimal function.

source on GitHub

__init__(booster)#
static _generate_classes(booster)#
get_objective()#

Returns the objective.

class mlprodict.onnx_conv.operator_converters.parse_lightgbm.WrappedLightGbmBoosterClassifier(wrapped)#

Bases: ClassifierMixin

Trick to wrap a LGBMClassifier into a class.

source on GitHub

__init__(wrapped)#
mlprodict.onnx_conv.operator_converters.parse_lightgbm.converter_lightgbm_concat(scope, operator, container)#

Converter for operator LightGBMConcat.

source on GitHub

mlprodict.onnx_conv.operator_converters.parse_lightgbm.lightgbm_parser(scope, model, inputs, custom_parsers=None)#

Agnostic parser for LightGBM Booster.

source on GitHub

mlprodict.onnx_conv.operator_converters.parse_lightgbm.shape_calculator_lightgbm_concat(operator)#

Shape calculator for operator LightGBMConcat.

source on GitHub