开始使用免费开始使用

自定义训练/测试索引

正如您在视频中所见,本章将聚焦一个真实数据集,把前面各章讨论的概念串联起来。

churn 数据集包含多家电信客户的数据,建模目标是预测哪些客户会取消服务(即流失)。

本章您将探索两类预测模型:glmnetrf。首先要做的是创建一个可复用的 trainControl 对象,以便稳定地对比这些模型。

本练习是课程的一部分

在 R 中使用 caret 的机器学习

查看课程

练习说明

churn_xchurn_y 已加载到您的工作区。

  • 使用 createFolds() 在目标变量 churn_y 上创建 5 折交叉验证。
  • 将这些折传给 trainControl(),创建一个可复用的 trainControl,用于比较模型。

交互式实操练习

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

# Create custom indices: myFolds
myFolds <- createFolds(___, k = 5)

# Create reusable trainControl object: myControl
myControl <- trainControl(
  summaryFunction = twoClassSummary,
  classProbs = TRUE, # IMPORTANT!
  verboseIter = TRUE,
  savePredictions = TRUE,
  index = ___
)
编辑并运行代码