如何设置类计数器/概率的阈值来标记R rpart中的“预测类”

时间:2015-05-13 07:50:22

标签: r decision-tree rpart

我使用rpart函数获取决策树,根据一组变量预测所有者/无所有者。下面是输出的摘录

Node number 2: 8 observations,    complexity param=0.08333333   predicted class=non-owner  expected loss=0.125  P(node) =0.3333333
class counts:     7     1    probabilities: 0.875 0.125 

从上面的输出可以看出,节点2被标记为“非所有者”的预测类,因为该节点中非所有者的类计数(概率为0.875)为7。 我想根据节点标记的默认阈值进行更改。详细说明,假设默认值为0.5 - 如果概率(非所有者的类别数超过数据点的1/2)是> 0.5然后节点被标记为非所有者 - 但我想将其更改为0.9,即,只有当非所有者的类计数超过9(满分10)时,该节点才应被标记为非所有者,否则所有者。

0 个答案:

没有答案