How to specify loss function when using 'fitrensemble' function?
7 views (last 30 days)
Show older comments
MathWorks Support Team
on 26 Mar 2021
Edited: MathWorks Support Team
on 29 Aug 2024
I am using "fitrensemble" function from MATLAB Statistics and Machine Learning Toolbox for a machine learning application. In the documentation, the default loss function is set to mean square error, how can I specify a different loss function?
Accepted Answer
MathWorks Support Team
on 29 Jul 2024
Edited: MathWorks Support Team
on 29 Aug 2024
"fitrensemble" function has different algorithms/solvers and the loss depends on the type of algorithm/solver you select for the training. Please run the below command in the command window of installed MATLAB R2019a version to read more about details on the ensemble algorithms:
>> web(fullfile(docroot, 'stats/ensemble-algorithms.html'))
For example, you can use the following code to specify algorithm/solver:
>> fitrensemble(model, 'Method', 'Bag')
Once a model is trained, you can evaluate the model with different loss functions via "resubLoss" function. However, this function cannot change the loss function for training.
Please run the below command in the command window of installed MATLAB R2019a version to get release specific documentation on "resubLoss" function:
>> web(fullfile(docroot, 'stats/regressionensemble.resubloss.html'))
Please follow the below link to search for the required information regarding the current release:
0 Comments
More Answers (0)
See Also
Categories
Find more on Regression Tree Ensembles in Help Center and File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!