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:
- 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:
- Returns:
Dataset.