使用自定义的 trainControl
现在您已经有了自定义的 trainControl 对象,使用 AUC 而非准确率来调参并评估 caret 模型就很容易了。您只需通过 trControl 参数把自定义的 trainControl 对象传给 train() 函数,例如:
train(<standard arguments here>, trControl = myControl)
这种写法能让您方便地保存一组自定义的建模参数,然后在多次调用 train() 时复用。在第 5 章中,您会大量使用这一技巧。
本练习是课程的一部分
在 R 中使用 caret 的机器学习
练习说明
- 使用
train()在Sonar数据中用所有其他变量预测Class(即Class ~ .)。应当使用glm模型(也就是将method设为"glm"),并使用您自定义的trainControl对象myControl。将结果保存到model。 - 将模型打印到控制台并查看输出。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Train glm with custom trainControl: model
# Print model to console