训练与测试数据集:数据划分

HR Analytics:用 R 预测员工流失

Anurag Gupta

People Analytics Practitioner

什么是模型?

 

HR Analytics:用 R 预测员工流失

HR Analytics:用 R 预测员工流失

使用 caret 划分数据

# Load caret
library(caret)

# Set seed
set.seed(567)

# Store row numbers for training dataset
index_train <- createDataPartition(emp_final$turnover, p = 0.5, list = FALSE)

# Create training dataset
train_set <- emp_final[index_train, ]

# Create testing dataset
test_set <- emp_final[-index_train, ]
HR Analytics:用 R 预测员工流失

让我们来练习!

HR Analytics:用 R 预测员工流失

Preparing Video For Download...