How to specify loss function when using 'fitrensemble' function?

7 views (last 30 days)
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
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:

More Answers (0)

Products


Release

R2019a

Community Treasure Hunt

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

Start Hunting!