拆分数据
在本练习中,您将使用 caret 包把数据拆分为训练集和测试集。接下来的课程中,您将用训练集构建逻辑回归模型,并用测试集验证这些模型。
本练习是课程的一部分
HR Analytics:用 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[___, ]