開始使用免費開始

切分資料

在這個練習中,你會使用 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[___, ]
編輯並執行程式碼