Skip to content

Supported Models

Linear Models

Model Task Description
linear_regression Regression Ordinary least squares
ridge Regression L2 regularization
lasso Regression L1 regularization (feature selection)
elasticnet Regression L1 + L2 regularization
logistic_regression Classification Linear classifier

Tree Models

Model Task Description
decision_tree Both Single decision tree

Bagging Models

Model Task Description
random_forest Both Bootstrap aggregating
extra_trees Both Extremely randomized trees

Boosting Models

Model Task Description
xgboost Both Gradient boosting (GPU support)
lightgbm Both Fast gradient boosting
gradient_boosting Both sklearn gradient boosting
hist_gradient_boosting Both Histogram-based (handles NaN)
adaboost Both Adaptive boosting

Instance-Based Models

Model Task Description
knn Both K-nearest neighbors
svm Both Support vector machine

Probabilistic Models

Model Task Description
naive_bayes Classification Gaussian naive Bayes

Ensemble Models

Model Task Description
stacking Both Stacking ensemble
voting Both Voting ensemble

Model Selection

open-mlpipe auto-selects models based on:

  • Task type (regression/classification)
  • Dataset size
  • Number of features
  • Data characteristics