在R语言中以派对语言显示rpart结果的功能

Question

我想创建一个函数来显示rpart 和partykit 的结果。 我使用以下代码将rpart 的结果与as.party 转换并显示了树结构。

library(partykit)
library(rpart)

result1 <- rpart(Species~.,data = iris)
plot(as.party(result1))

所以，我在函数中定义了rpart和as.party并执行了。

rpart_party <- function(formula, data){
  result1 <- rpart(formula = formula ,data = data)
  return(as.party(result1))
}
plot(rpart_party(Species~., data = iris))

但是，发生了以下错误。

Error in eval(predvars, data, env) : 
  invalid 'envir' argument of type 'closure'

请告诉我如何创建一个函数，将rpart 的结果转换为as.party 并显示树的结构。

Answer 1

您需要将您对公式的评估包含在substitute(...) 之类的内容中，以防止将模型调用设置为“公式”而不是“物种~”。等等。

rpart_party <- function(formula, data){
  result1 <- rpart(formula = substitute(formula), data = data)
  return(as.party(result1))
}