Main Content

Regression Trees

Binary decision trees for regression

To interactively grow a regression tree, use the Regression Learner app. For greater flexibility, grow a regression tree using fitrtree at the command line. After growing a regression tree, predict responses by passing the tree and new predictor data to predict.

Apps

Regression LearnerTrain regression models to predict data using supervised machine learning

Blocks

RegressionTree PredictPredict responses using regression tree model (Since R2021a)

Functions

expand all

fitrtreeFit binary decision tree for regression
compactCompact regression tree
pruneProduce sequence of regression subtrees by pruning
limeLocal interpretable model-agnostic explanations (LIME) (Since R2020b)
nodeVariableRangeRetrieve variable range of decision tree node (Since R2020a)
partialDependenceCompute partial dependence (Since R2020b)
plotPartialDependenceCreate partial dependence plot (PDP) and individual conditional expectation (ICE) plots
predictorImportanceEstimates of predictor importance for regression tree
surrogateAssociationMean predictive measure of association for surrogate splits in regression tree
shapleyShapley values (Since R2021a)
viewView regression tree
crossvalCross-validated decision tree
cvlossRegression error by cross validation
kfoldfunCross-validate function for regression
kfoldPredictPredict responses for observations in cross-validated regression model
kfoldLossLoss for cross-validated partitioned regression model
lossRegression error
resubLossRegression error by resubstitution
predictPredict responses using regression tree
resubPredictPredict resubstitution response of tree
gatherGather properties of Statistics and Machine Learning Toolbox object from GPU (Since R2020b)

Classes

RegressionTreeRegression tree
CompactRegressionTreeCompact regression tree
RegressionPartitionedModelCross-validated regression model

Topics