Random forest:I want to know the probabilities of the classifications.

1 view (last 30 days)
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.

Answers (0)

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!