MATLAB Answers

How can I use Grid search on these parameters?

28 views (last 30 days)
Yooya yooya
Yooya yooya on 11 Dec 2020
Answered: Puru Kathuria on 22 Dec 2020
How can I use Grid search on following parameters?
-MaxNumSplits
-MinLeafSize
-SplitCriterion
-NumVariablesTosample
% decision tree model
t = ClassificationTree.fit(Xtrain,Ytrain,'CategoricalPredictors',catPred);

Answers (1)

Puru Kathuria
Puru Kathuria on 22 Dec 2020
The following post explains how to do a grid search for SVM. You can do the same for decision tree by just changing the parameters.
Also, this link contains a function that does grid search on a neural network, you can also take inspiration from this function to perform the same on your decision tree.
The main idea is to use ndgrid and pass the parameters you want to optimize to ndgrid and form pairs to perform the exhaustive search using ndgrid.

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!