开始使用免费开始使用

拆分数据

在本练习中,您将使用 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[___, ]
编辑并运行代码