如何改变二值分类的预测阈值?

phivuphivu 成员职位:34大师
2019年11月编辑 帮助

嗨RapidMiner,

我正在用不同的方法构建一个二元分类器。但我注意到,不同的方法对置信度(yes)值使用不同的阈值来决定预测是yes还是no。例如,SVM使用0.5作为阈值。然而,梯度增强树(GBT)似乎将阈值设置为0.1。所以我想问一下是否有办法改变GBT的这个阈值,因为这会影响性能运算符中计算的精度(如果我可以将GBT的阈值设置为0.5,我会得到更高的精度)。

我在这条消息中附上了这个过程以及训练好的模型和测试数据。如果您运行该流程,您将看到预测结果中的一些记录(例如行号368,471,529,573)置信度(yes) < 0.2,但预测是yes。

非常感谢你的帮助!

最好的
phivu

标记:

最佳答案

答案

  • phivuphivu 成员职位:34大师

    这个效果很好,谢谢你,马丁!

    phivu

登录注册置评。