sparklightautoml.automl

The main module, which includes the SparkAutoML class, blenders and ready-made presets.

SparkAutoML

Class for compile full pipeline of AutoML task.

Presets

Presets for end-to-end model training for special tasks.

base.SparkAutoMLPreset

Basic class for automl preset.

tabular_presets.SparkTabularAutoML

Spark version of TabularAutoML.

Blenders

SparkBlender

Basic class for blending.

SparkBestModelSelector

Select best single model from level.

SparkMeanBlender

Simple average level predictions.

SparkWeightedBlender

Weighted Blender based on coord descent, optimize task metric directly.