为树模型确定 R 中特定结果的概率答案

【问题标题】：Determine probability of specific outcome in R for Tree Model为树模型确定 R 中特定结果的概率
【发布时间】：2018-07-27 20:57:57
【问题描述】：

对于我使用 method = 'poisson' 创建的树模型，在给定 new.data (X1=4,X2=4,X3=4) 的情况下，我如何找到获得 Y=0 结果的概率？具体代码如下：

tree.pois.cp<-rpart(Y ~ X1 + X2 + X3, data = data, method = 'poisson', control = rpart.control(cp = 1.1034e-02))

我用下面的代码做同样的事情，但使用了我的负二项式模型：

pred.y.nb<-predict(nb, newdata = new.data, type = "response")
prob0.nb<-dnbinom(0, mu=pred.y.nb, size=nb$theta)
prob0.nb
#this is my answer for probability of Y=0 given my negative binom model

（大声说出这个问题来帮助我：How to calculate the predicted probability of negative binomial regression model?）

我尝试对我的树模型 tree.pois.cp 使用相同的代码：

pred.y.pois.cp<-predict(tree.pois.cp, newdata = new.data, type = "response")

但我收到此错误：

Error in match.arg(type) : 'arg' should be one of “vector”, “prob”, “class”, “matrix”

感谢您的帮助！

【问题讨论】：

标签： r

【解决方案1】：

请阅读rpart 文档。没有 type = "response" 用于预测 rpart 对象。你可以试试下面的代码：

data<-data.frame(Y=as.character(rpois(n = 20000,.2)),X1=sample(1:4,20000,replace = T),X2=sample(1:4,20000,replace = T),X3=sample(1:4,20000,replace = T),X4=sample(1:4,20000,replace = T))

tree.pois.cp<-rpart(Y ~ X1 + X2 + X3, data = data, method = 'class')
new.data<- data.frame(Y="0",X1=4,X2=4,X3=4,X4=4)
pred.y.pois<-predict(tree.pois.cp, newdata = new.data, type = "prob")
pred.y.pois

【讨论】：