我使用R中的tree
包来制作决策树,但是为了做模型的ROC曲线我需要预测的概率而我找不到获取它的方法,我只是得到预测响应。我的代码如下,其中Status是因变量(“好”或“坏”),我的数据集称为“dd”,“learn”是学习的子集,“-learn”表示测试的子集
tree.dd=tree(Status~.,dd,subset=learn)
tree.test.prob <- predict (tree.dd, newdata = dd[-learn,])
pred = prediction(tree.test.prob, dd[-learn,]$Status)
我需要预测概率而不是反应。