使用树包

时间:2018-04-12 23:18:07

标签: r classification decision-tree roc

我使用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)

我需要预测概率而不是反应。

0 个答案:

没有答案