Функция для отображения результата rpart на партийном языке на языке R

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))
}