R使用caretEnsemble包中的train函数训练模型时遇到的错误
使用Caret包中的trainControl训练分类模型时遇到的错误
·
使用caretEnsemble包中的trainControl和train函数训练模型,使用的数据集为标签为label的分类数据,代码如下:
control <- trainControl(method="repeatedcv", number=5, repeats=1, savePredictions=TRUE, classProbs=TRUE)
rf<- train(label ~ ., data = train_data,
method = "rf",
trControl = control)
rf
运行后发生错误
In addition: Warning messages:
1: In train.default(x, y, weights = w, ...) :
You are trying to do regression and your outcome only has two possible values Are you trying to do classification? If so, use a 2 level factor as your outcome column.
2: In train.default(x, y, weights = w, ...) :
cannnot compute class probabilities for regression
解决方法:将数据集标签列的数据由0,1改为yes,no,数值类型的数据被R默认是做回归,改为两种类别的字符即可继续运行。
更多推荐
所有评论(0)