AutoML

class lightautoml.automl.base.AutoML(reader, levels, timer=None, blender=None, skip_conn=False, return_all_predictions=False)[source]

Bases: object

Class for compile full pipeline of AutoML task.

AutoML steps:

  • Read, analyze data and get inner LAMLDataset from input dataset: performed by reader.

  • Create validation scheme.

  • Compute passed ml pipelines from levels. Each element of levels is list of MLPipelines prediction from current level are passed to next level pipelines as features.

  • Time monitoring - check if we have enough time to calc new pipeline.

  • Blend last level models and prune useless pipelines to speedup inference: performed by blender.

  • Returns prediction on validation data. If crossvalidation scheme is used, out-of-fold prediction will returned. If validation data is passed it will return prediction on validation dataset. In case of cv scheme when some point of train data never was used as validation (ex. timeout exceeded or custom cv iterator like TimeSeriesIterator was used) NaN for this point will be returned.

Example

Common usecase - create custom pipelines or presets.

>>> reader = SomeReader()
>>> pipe = MLPipeline([SomeAlgo()])
>>> levels = [[pipe]]
>>> automl = AutoML(reader, levels, )
>>> automl.fit_predict(data, roles={'target': 'TARGET'})
__init__(reader, levels, timer=None, blender=None, skip_conn=False, return_all_predictions=False)[source]
Parameters

Note

There are several verbosity levels:

  • 0: No messages.

  • 1: Warnings.

  • 2: Info.

  • 3: Debug.

fit_predict(train_data, roles, train_features=None, cv_iter=None, valid_data=None, valid_features=None, verbose=0)[source]

Fit on input data and make prediction on validation part.

Parameters
Return type

LAMLDataset

Returns

Predicted values.

predict(data, features_names=None, return_all_predictions=None)[source]

Predict with automl on new dataset.

Parameters
  • data (Any) – Dataset to perform inference.

  • features_names (Optional[Sequence[str]]) – Optional features names, if cannot be inferred from train_data.

  • return_all_predictions (Optional[bool]) – if True, returns all model predictions from last level

Return type

LAMLDataset

Returns

Dataset with predictions.

collect_used_feats()[source]

Get feats that automl uses on inference.

Return type

List[str]

Returns

Features names list.

collect_model_stats()[source]

Collect info about models in automl.

Return type

Dict[str, int]

Returns

Dict with models and its runtime numbers.