嗨,我的名字是 Abhi,我正在使用插入符号来构建基于 gbm 树的模型。但是,我不想使用准确性,而是使用 roc 作为我的指标
这是我到目前为止的代码
myTuneGrid <- expand.grid(n.trees = 500,interaction.depth = 11,shrinkage = 0.1)
fitControl <- trainControl(method = "repeatedcv", number = 7,repeats = 1, verboseIter = FALSE,returnResamp = "all",classProbs = TRUE)
myModel <- train(Cover_Type ~ .,data = modelData,method = "gbm",trControl = fitControl,tuneGrid = myTuneGrid,metric='roc')
但是,当我运行此代码时,我会收到警告
Warning message:
In train.default(x, y, weights = w, ...) :
The metric "roc" was not in the result set. Accuracy will be used instead.
如何强制我的模型使用 roc 而不是准确性。我在这里做错了什么?