Skip to content

Latest commit

 

History

History
53 lines (41 loc) · 4.85 KB

classification.md

File metadata and controls

53 lines (41 loc) · 4.85 KB
description
Technique that assigns new observations to categorical labels or classes based on a model built from labeled training data.

Classification

We currently support classification algorithms from scikit-learn, XGBoost, and LightGBM.

Gradient Boosting

Algorithm Classification
xgboost XGBClassifier
xgboost_random_forest XGBRFClassifier
lightgbm LGBMClassifier
catboost CatBoostClassifier

Scikit Ensembles

Algorithm Classification
ada_boost AdaBoostClassifier
bagging BaggingClassifier
extra_trees ExtraTreesClassifier
gradient_boosting_trees GradientBoostingClassifier
random_forest RandomForestClassifier
hist_gradient_boosting HistGradientBoostingClassifier

Support Vector Machines

Algorithm Classification
svm SVC
nu_svm NuSVC
linear_svm LinearSVC

Linear Models

Algorithm Classification
linear LogisticRegression
ridge RidgeClassifier
stochastic_gradient_descent SGDClassifier
perceptron Perceptron
passive_aggressive PassiveAggressiveClassifier

Other

Algorithm Classification
gaussian_process GaussianProcessClassifier