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

1 visualizzazione (ultimi 30 giorni)
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.

Risposte (0)

Categorie

Scopri di più su Get Started with Statistics and Machine Learning Toolbox in Help Center e File Exchange

Community Treasure Hunt

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

Start Hunting!

Translated by