Random forest:I want to know the probabilities of the classifications.
1 view (last 30 days)
Show older comments
Hello. I'm a Japanese graduate school student and bigginer of machine learning.
I'm creating a random forest model to classify patients into complication group and nomal group after the treatment.
Mdl = TreeBagger(NumTrees,X,Y,Name,Value)
Prediction = predict(Mdl,testdata)
The model answers 0 or 1 after the training.
I want to know the probability of the classification.
example)
case1 the complication occurs with 23% →0
case2 the complication occurs with 67% →1
I want to know these [23% 67%] of training data (OOB) or test data, due to classify into lowrisk, moderaterisk and high risk group(not just 0 or 1). Please tell me your advice or the appropriate code.
0 Comments
Answers (0)
See Also
Categories
Find more on Get Started with Statistics and Machine Learning Toolbox 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!