自定义模型评估
在迭代不同模型时,创建自定义评估函数会非常方便。yardstick 包中的 metric_set() 函数可以帮助您实现这一点。
请定义一个函数,返回 roc_auc、accuracy、sens(敏感度)和 spec(特异度),并用它来评估您的模型。
增强后的数据框 lr_aug 已经加载,随时可用。
本练习是课程的一部分
R 中的特征工程
练习说明
- 定义一个自定义评估函数,返回
roc_auc、accuracy、sens和spec。 - 使用您新建的函数在
lr_aug上评估模型,得到您刚才选择的这些度量。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Define a custom assessment function
class_evaluate <- ___(___, ___, sens, ___)
# Assess your model using your new function
___(___, truth = Attrition,
estimate = .pred_class,
.pred_No)