Caret:选择未定义的列

Question

我一直试图让下面的代码在插入符号中运行,但得到错误.谁能告诉我怎么麻烦拍它.

[.data.frame(data,,lvls [1])出错:选择了未定义的列

library(tidyverse)
library(caret)

mydf <- iris

mydf <- mydf %>% 
  mutate(tgt = as.factor(ifelse(Species == 'setosa','Y','N'))) %>% 
  select(everything(), -Species)

trainIndex <- createDataPartition(mydf$tgt, p = 0.75, times = 1, list = FALSE)
train <- mydf[trainIndex,]
test <- mydf[-trainIndex,]

fitControl <- trainControl(method = 'repeatedcv',
                       number = 10,
                       repeats = 10,
                       allowParallel = TRUE,
                       summaryFunction = twoClassSummary)

fit_log <- train(tgt~.,
             data = train,
             method = "glm",
             trControl = fitControl,
             family = "binomial")

Answer 1

您需要classProbs = TRUE在控制功能中使用.ROC曲线基于类概率,错误是汇总函数未找到这些列.