r - Caret/CaretEnsemble：错误的回归模型类型

Question

我正在尝试构建一组机器学习算法，第一个使用提升算法，第二个使用堆叠算法。当我尝试创建模型时，出现以下错误：

Error: wrong model type for regression

这是我的代码：

library(mlbench)
library(caret)
library(caretEnsemble)

#Remove Null Values
Data <- Data[complete.cases(Data), ]    

control <- trainControl(method="repeatedcv", number=10, repeats=3, 
savePredictions=TRUE, classProbs=TRUE)
algorithmList <- c('lda', 'rpart', 'glm', 'knn', 'svmRadial')
set.seed(seed)
models <- caretList(Setup~Age+Balance+Dcny2Rcvd+PCA, 
data=data, trControl=control, methodList=algorithmList)
results <- resamples(models)
summary(results)
dotplot(results)

这是我的数据示例：

Number    Setup      R         D2R     A        B       PCA   
123        1        381        6618    25      2500      2
124        0        200        100     30      3500      1
125        0        100        62      18      4000      1
126        1        202        75      21      1000      5
127        0        95         60      22      387       4

问题：为什么我会收到此错误？

对于以下内容，我也遇到了同样的错误：

fit.c50 <- train(Setup~A+B+D2R+PCA+R, data=Data, trControl=control)

标签： rregressiondata-sciencer-caret

Answer 1

如果你想用 进行分类caret，你的结果Setup应该是一个因子变量。由于它是一个数字，它认为您正在尝试使用分类模型进行回归。