description |
---|
Technique that assigns new observations to categorical labels or classes based on a model built from labeled training data. |
We currently support classification algorithms from scikit-learn, XGBoost, and LightGBM.
Algorithm | Classification |
---|---|
xgboost |
XGBClassifier |
xgboost_random_forest |
XGBRFClassifier |
lightgbm |
LGBMClassifier |
catboost |
CatBoostClassifier |
Algorithm | Classification |
---|---|
ada_boost |
AdaBoostClassifier |
bagging |
BaggingClassifier |
extra_trees |
ExtraTreesClassifier |
gradient_boosting_trees |
GradientBoostingClassifier |
random_forest |
RandomForestClassifier |
hist_gradient_boosting |
HistGradientBoostingClassifier |
Algorithm | Classification |
---|---|
svm |
SVC |
nu_svm |
NuSVC |
linear_svm |
LinearSVC |
Algorithm | Classification |
---|---|
linear |
LogisticRegression |
ridge |
RidgeClassifier |
stochastic_gradient_descent |
SGDClassifier |
perceptron |
Perceptron |
passive_aggressive |
PassiveAggressiveClassifier |
Algorithm | Classification |
---|---|
gaussian_process |
GaussianProcessClassifier |