The AI & Analytics Engine offers a variety of high-quality machine learning model templates.
The model templates are grouped according to the ML approach;
Some templates appear across segmented lists.Supervised Learning
Supervised machine learning is an approach using labeled datasets. These datasets are designed to train or “supervise” algorithms into classifying data or predicting outcomes accurately. Using paired inputs and labels, these models can be properly evaluated and improved with more data available over time.
Regression Model Templates/Algorithms
-
AdaBoost regressor
-
Bayesian ridge regression
-
Decision tree regressor
-
Extra-trees regressor
-
Gradient boosting regressor
-
K-Nearest neighbors regressor
-
LightGBM regressor
-
Linear regression (with support of GPU, multi-GPU)
-
Ridge regression (with support of multi-GPU)
-
Mini-batch SGD (stochastic gradient descent) regressor (with support of GPU)
-
Random forest regressor (with support of GPU, multi-GPU)
-
XGBoost Regressor (with support of GPU, multi-node, multi-GPU)
Classification Model Templates/Algorithms
-
AdaBoost classifier
-
Random forest classifier (with support of GPU, multi-GPU)
-
K-Nearest neighbors classifier
-
Logistic regression (with support of GPU)
-
Mini-batch SGD (stochastic gradient descent) classifier (with support of GPU)
-
Decision tree classifier
-
Extra-trees classifier
-
Gradient boosting classifier
-
LightGBM classifier
-
Gaussian Naive Bayes
-
XGBoost classifier (with support of GPU, multi-node, multi-GPU)
Time Series Forecasting
Time series forecasting is the use of algorithms (model templates) to predict future values based on historical time-stamped data. The Engine supports the following algorithms (model templates):
-
ARIMA (Autoregressive Integrated Moving Average)
-
Prophet
-
ETS (Error, Trend, Seasonality)
-
TBATS (Trigonometric regressors, Box-Cox transformations, ARMA errors, Trend, Seasonality)
-
Theta models
Want to learn more about machine learning problem types? Read our helpful blog article "What is Machine Learning Part 2"