r - 错误 : `i` must have one dimension, 不是 2. 在 R 中的 CreateDataPartition

Question

我正在运行一段我以前成功运行过多次的代码，现在我遇到了一个我不明白的奇怪错误:

### Data Splitting

trainIndex <- createDataPartition(dataset$isTerminated, p = .8, 
                                  list = FALSE, 
                                  times = 1)

head(trainIndex)

# Step 2: Create the training  and testing datasets

testData  <- dataset[-trainIndex, ]
trainingData <- dataset[trainIndex, ]

testData <- dataset[-trainIndex, ]工作正常，但当我运行时 trainingData <- dataset[trainIndex, ]给我一个错误 Error:我 must have one dimension, not 2.

我已经根据 https://topepo.github.io/caret/data-splitting.html 检查了代码它匹配。有没有其他人遇到过它并找到了解决方案？

Answer 1

检查数据的类别。如果是 tibble，将其转换为 data.frame。