Question about classification probability

I would like to do classification (value: A or B), and I try to use Gradient Boosted or Random Forest to perform the classification.

After model completion, the predicted classification (“A” or “B”) are shown. However, I would like to generate the probability of each forecast, is there any way to do it?

Thanks
Lawson

Hi @Lawson -

In the predictor node for the model you’re using, make sure to check “Append Individual class probabilities”. This will give you what you’re looking for.

(Note that in KNIME 3.6, the prediction confidence is now generated by default, which is the maximum of the individual class probabilities.)

2 Likes

Thanks

1 Like

This topic was automatically closed 7 days after the last reply. New replies are no longer allowed.