开始使用免费开始使用

自定义 trainControl

葡萄酒质量数据集是一个回归问题,但现在您将处理一个分类问题。这个数据集是基于多年前 Kaggle 上 "don't overfit" 竞赛的模拟数据。

分类问题比回归问题稍复杂,因为若要用 AUC 指标对模型进行排序,您需要为 train() 函数提供自定义的 summaryFunction。请先像上一章那样创建一个自定义的 trainControl。务必将 classProbs = TRUE 设为 TRUE,否则用作 summaryFunctiontwoClassSummary 将无法工作。

本练习是课程的一部分

在 R 中使用 caret 的机器学习

查看课程

练习说明

使用 trainControl 函数为分类创建名为 myControl 的自定义 trainControl

  • 使用 10 折交叉验证。
  • summaryFunction 中使用 twoClassSummary
  • 一定要设置 classProbs = TRUE

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Create custom trainControl: myControl
myControl <- trainControl(
  method = "cv", 
  number = ___,
  summaryFunction = ___,
  classProbs = ___, # IMPORTANT!
  verboseIter = TRUE
)
编辑并运行代码