使用带自定义权重的gbm时,插入符号中的train功能出错

时间:2015-08-20 16:16:29

标签: r parameters r-caret gbm

我正在尝试使用自定义权重参数和接收错误在插入符包(RStudio)中使用train函数为gbm执行参数调整。错误是

Error in {: task 1674 failed - inputs must be factors

原始数据集包含1649个观察结果,并使用60/40分割将其分为训练和测试集。调整参数使用trainControl和网格来定义,以尝试不同的参数值。

对于是/否类,列权重包含1或10,并且是数据框中值的数字向量。主函数调用如下:

model <- train(train[,predictors],train[,class], method="gbm", weights=df$weights, trControl=trainControl obj, tuneGrid=Grid obj, metric="ROC")

df $ weights是每个观察的权重向量。由于类不平衡,使用度量“ROC”。在运行模型之前,该类被转换为包含yes,no值的因子。此外,如果我没有提到权重参数,那么函数工作正常。

如果有人能够解释如何克服这个错误,或者有人遇到同样的麻烦以及他们如何纠正错误,我会很感激。

感谢。

0 个答案:

没有答案