lightautoml.ml_algo

Models used for machine learning pipelines.

Base Classes

MLAlgo	Abstract class for machine learning algorithm.
TabularMLAlgo	Machine learning algorithms that accepts numpy arrays as input.

Linear Models

LinearLBFGS	LBFGS L2 regression based on torch.
LinearL1CD	Coordinate descent based on sklearn implementation.
TorchModel	Neural net for tabular datasets.

Boosted Trees

BoostLGBM	Gradient boosting on decision trees from LightGBM library.
BoostCB	Gradient boosting on decision trees from catboost library.

Neural Networks

MLP	Realisation of 'mlp' model.
DenseLightModel	Realisation of 'denselight' model.
DenseModel	Realisation of 'dense' model.
ResNetModel	The ResNet model from https://github.com/Yura52/rtdl.
SNN	Realisation of 'snn' model.

WhiteBox

WbMLAlgo

WhiteBox - scorecard model.