【发布时间】:2015-04-22 14:36:44
【问题描述】:
虽然我定义了 target <- factor(train$target, levels = c(0, 1)),但下面给出的代码提供了这个错误:
cut.default(y, unique(quantile(y, probs = seq(0, 1, length = cut))), : 无效的间隔数另外:警告 消息:1:在 train.default(x, y, weights = w, ...) 中:不能 计算回归的类别概率
这是什么意思以及如何解决这个问题?
gbmGrid <- expand.grid(n.trees = (1:30)*10,
interaction.depth = c(1, 5, 9),
shrinkage = 0.1)
fitControl <- trainControl(method = "repeatedcv",
number = 5,
repeats = 5,
verboseIter = FALSE,
returnResamp = "all",
classProbs = TRUE)
target <- factor(train$target, levels = c(0, 1))
gbm <- caret::train(target ~ .,
data = train,
#distribution="gaussian",
method = "gbm",
trControl = fitControl,
tuneGrid = gbmGrid)
prob = predict(gbm, newdata=testing, type='prob')[,2]
【问题讨论】:
-
train中的target <- factor(train$target,...)是什么? -
尝试在您的
trainControl函数中添加summaryFunction = twoClassSummary,看看它是否有效。