使用 tidymodels 创建并绘制 ROC 曲线

Question

在 tidymodels 中尝试分类袋装树，一切都很好，直到我开始创建和绘制 ROC 曲线。然后我得到一个

# Error in 'roc_curve()': ! Can't rename variables in this context. 

我的 csv 和我编写的脚本位于 here

对训练集的预测没有产生错误

predictions<- predict(model_bagged, bank_train, type= "prob") %>% bind_cols(bank_train)

输出中有一个 .pred_yes 列，所以我不确定错误来自哪里或它意味着什么。

Answer 1

roc_curve()

函数不使用

estimate

参数，而是需要您将未命名的估计值的列传递给

...

，如下所示：

roc_curve(predictions,
          truth = still_customer,
          .pred_yes)

所有类概率指标都是这种情况