自訂你的模型評估
在不同模型之間反覆實驗時,自訂評估函式相當方便。yardstick 套件的 metric_set() 函式可以幫上忙。
定義一個會回傳 roc_auc、accuracy、sens(靈敏度,sensitivity)與 spec(特異度,specificity)的方法,並用它來評估你的模型。
增廣後的資料框 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)