The Altair Community is migrating to a new platform to provide a better experience for you. The RapidMiner Community will merge with the Altair Community at the same time. In preparation for the migration, both communities are on read-only mode from July 15th - July 24th, 2024. Technical support via cases will continue to work as is. For any urgent requests from Students/Faculty members, please submit the form linked here.
Options

How to change prediction threshold for binary classification?

phivuphivu Member Posts: 34 Guru
edited November 2019 in Help

Hi RapidMiner,

 

I'm building a binary classifier using different methods. But i notice that different methods use different thresholds on the confidence(yes) values to decide if the prediction is yes or no. For example, SVM uses 0.5 as the threshold. However, Gradient Boosted Trees (GBT) seems to set the threshold as 0.1. So i'd like to ask if there's a way to change this threshold for GBT, cause this will affect the accuracy computed in the Performance operator (if i could set the threshold as 0.5 for GBT, i will get a higher accuracy).

 

I attached the process together with the trained models and testing data in this message. If you run the process, you will see some records in the prediction result (e.g. row number 368, 471, 529, 573) have confidence(yes) < 0.2 but the predictions are yes.

 

Thanks a lot for your help!

 

Best,
phivu

Tagged:

Best Answer

Answers

  • Options
    phivuphivu Member Posts: 34 Guru

    This works great, thank you Martin!

     

    phivu

Sign In or Register to comment.