module onnx_conv.operator_converters.parse_lightgbm
#
Short summary#
module mlprodict.onnx_conv.operator_converters.parse_lightgbm
Parsers for LightGBM booster.
Classes#
class |
truncated documentation |
---|---|
Mocked lightgbm. |
|
A booster can be a classifier, a regressor. Trick to wrap it in a minimal function. |
|
Trick to wrap a LGBMClassifier into a class. |
Functions#
function |
truncated documentation |
---|---|
Converter for operator LightGBMConcat. |
|
Agnostic parser for LightGBM Booster. |
|
Shape calculator for operator LightGBMConcat. |
Static Methods#
staticmethod |
truncated documentation |
---|---|
Methods#
method |
truncated documentation |
---|---|
Returns default values for common attributes. |
|
mock dump_model method |
|
Returns binary features names. |
|
Returns the objective. |
Documentation#
Parsers for LightGBM booster.
- class mlprodict.onnx_conv.operator_converters.parse_lightgbm.MockWrappedLightGbmBoosterClassifier(tree)#
Bases:
WrappedLightGbmBoosterClassifier
Mocked lightgbm.
- __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.
- __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.
- __init__(wrapped)#
- mlprodict.onnx_conv.operator_converters.parse_lightgbm.converter_lightgbm_concat(scope, operator, container)#
Converter for operator LightGBMConcat.
- mlprodict.onnx_conv.operator_converters.parse_lightgbm.lightgbm_parser(scope, model, inputs, custom_parsers=None)#
Agnostic parser for LightGBM Booster.
- mlprodict.onnx_conv.operator_converters.parse_lightgbm.shape_calculator_lightgbm_concat(operator)#
Shape calculator for operator LightGBMConcat.