r - カスタム指標の標準偏差を使用してキャレットで調整パラメーターを選択する

Question

カスタムフィッティングメトリックでキャレットを使用していますが、このメトリックだけでなく、信頼区間の下限も最大化する必要があります。だから私はのようなものを最大化したいと思いmean(metric) - k * stddev(metric)ます。これを手動で行う方法は知っていますが、この関数を使用して最適なパラメーターを自動的に選択するようにキャレットに指示する方法はありますか?

Answer 1

train 関数のキャレットのヘルプには、より基本的な例があります。

madSummary <- function (data,
                        lev = NULL,
                        model = NULL) {
  out <- mad(data$obs - data$pred, 
             na.rm = TRUE)  
  names(out) <- "MAD"
  out
}

robustControl <- trainControl(summaryFunction = madSummary)
marsGrid <- expand.grid(degree = 1, nprune = (1:10) * 2)

earthFit <- train(medv ~ .,
                  data = BostonHousing, 
                  method = "earth",
                  tuneGrid = marsGrid,
                  metric = "MAD",
                  maximize = FALSE,
                  trControl = robustControl)