Tree-based algorithms.
This module provides decision trees and tree ensembles for classification and regression tasks.
Algorithms
-
DecisionTreeClassifier: CART decision tree classifier.
-
DecisionTreeRegressor: CART decision tree regressor.
-
DecisionStumpClassifier: One-level decision tree (weak learner).
-
RandomForestClassifier: Ensemble of random trees for classification.
-
RandomForestRegressor: Ensemble of random trees for regression.