WBPipeline

class lightautoml.pipelines.ml.whitebox_ml_pipe.WBPipeline(whitebox)[source]

Bases: MLPipeline

Special pipeline to handle WhiteBox model.

property whitebox

Return first whitebox model.

__init__(whitebox)[source]

Create WhiteBox MLPipeline.

Parameters:

whitebox (Union[WbMLAlgo, Tuple[WbMLAlgo, ParamsTuner]]) – WhiteBox model.

fit_predict(train_valid)[source]

Fit WhiteBox.

Parameters:

train_valid (TrainValidIterator) – Classic cv-iterator.

Return type:

NumpyDataset

Returns:

Dataset.

predict(dataset, report=False)[source]

Predict WhiteBox.

Additional report param stands for WhiteBox report generation.

Parameters:
  • dataset (PandasDataset) – Dataset of text features.

  • report (bool) – Flag if generate report.

Return type:

NumpyDataset

Returns:

Dataset.