module onnx_conv.parsers.parse_lightgbm

Short summary

module mlprodict.onnx_conv.parsers.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.parsers.parse_lightgbm.MockWrappedLightGbmBoosterClassifier(tree)

Bases: mlprodict.onnx_conv.parsers.parse_lightgbm.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.parsers.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.parsers.parse_lightgbm.WrappedLightGbmBoosterClassifier(wrapped)

Bases: sklearn.base.ClassifierMixin

Trick to wrap a LGBMClassifier into a class.

source on GitHub

__init__(wrapped)

mlprodict.onnx_conv.parsers.parse_lightgbm.converter_lightgbm_concat(scope, operator, container)

Converter for operator LightGBMConcat.

source on GitHub

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

Agnostic parser for LightGBM Booster.

source on GitHub

mlprodict.onnx_conv.parsers.parse_lightgbm.shape_calculator_lightgbm_concat(operator)

Shape calculator for operator LightGBMConcat.

source on GitHub