切分資料
在這個練習中,你會使用 caret 套件將資料分為訓練集與測試集。接下來的課程中,你會用訓練集建立邏輯斯迴歸模型,並用測試集驗證這些模型。
本練習屬於課程
HR 分析:以 R 預測員工流失
練習說明
- 載入
caret套件。 - 設定隨機種子為 567,並建立資料分割,將資料集
emp_final依 70%/30% 劃分為訓練/測試。 - 建立訓練資料集:從
emp_final中選取儲存在index_train的列號。 - 將
emp_final中剩餘的觀測分配到測試集。
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
# Load caret
___
# Set seed of 567
___
# Store row numbers for training dataset: index_train
index_train <- ___(emp_final$turnover, p = ___, list = FALSE)
# Create training dataset: train_set
train_set <- emp_final[___, ]
# Create testing dataset: test_set
test_set <- emp_final[___, ]