ProbabilityColsTransformer

class sparklightautoml.transformers.base.ProbabilityColsTransformer(probability_cols=None, num_classes=0)[source]

Bases: Transformer, DefaultParamsWritable, DefaultParamsReadable

Converts probability columns values from ONNX model format to LGBMCBooster format