Model types

Model types that can be computed. Lists taken from Pycaret Documentation

Regression models:

‘lr’ - Linear Regression

‘lasso’ - Lasso Regression

‘ridge’ - Ridge Regression

‘en’ - Elastic Net

‘lar’ - Least Angle Regression

‘llar’ - Lasso Least Angle Regression

‘br’ - Bayesian Ridge

‘kr’ - Kernel Ridge

‘svm’ - Support Vector Regression

‘knn’ - K Neighbors Regressor

‘dt’ - Decision Tree Regressor

‘rf’ - Random Forest Regressor

‘et’ - Extra Trees Regressor

‘ada’ - AdaBoost Regressor

‘gbr’ - Gradient Boosting Regressor

‘xgboost’ - Extreme Gradient Boosting

‘lightgbm’ - Light Gradient Boosting Machine

‘catboost’ - CatBoost Regressor

Classification models:

‘kmeans’ - K-Means Clustering

‘ap’ - Affinity Propagation

‘meanshift’ - Mean shift Clustering

‘sc’ - Spectral Clustering

‘hclust’ - Agglomerative Clustering

‘dbscan’ - Density-Based Spatial Clustering

‘optics’ - OPTICS Clustering

‘birch’ - Birch Clustering

‘kmodes’ - K-Modes Clustering