開始使用免費開始

切分訓練與測試資料集

訓練模型的第一步,是將資料分成訓練集與測試集。tidymodels 套件能讓這件事變得很簡單。預留一個測試資料集,可以讓你在模型從未看過的資料上評估已訓練的模型。

你將使用員工醫療產業離職資料,內容包含某醫療公司的員工資訊,以及他們是否離職。此資料已存在 attrition_df 中。目標變數是 Attrition

tidayversetidymodels 套件都已為你載入。

本練習屬於課程

R 的降維

檢視課程

練習說明

  • 以 80% 的資料作為訓練集來初始化資料切分,並根據目標變數 Attrition 進行分層抽樣。
  • 擷取訓練資料集並存到 train
  • 擷取測試資料集並存到 test

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

# Initialize the split
split <- ___(___, ___ = ___, strata = ___)

# Extract training set
train <- ___ %>% ___()

# Extract testing set
test <- ___ %>% ___()
編輯並執行程式碼