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.

WhiteBox

WbMLAlgo

WhiteBox - scorecard model.