r语言员工离职_使用R机器学习进行员工离职预测系列(一）

最近一直觉得，其实机器学习的门槛并不高，以R语言的角度，甚至稍微学过一点，就可以针对相关数据进行各种算法模型的建立和测试。

而真正有难度的地方一是算法优化部分，二是和对于模型评价的部分，这两个部分往往涉及更多的知识点。

本文用的数据集是kaggle上的员工离职数据集，大家可以自行下载，下面是用R选择逻辑回归算法建模预测的过程。

首先加载数据

hrdata

看数据结构，预测变量是left，就是员工是否离职

str(hrdata)

'data.frame':14999 obs. of 10 variables:

$ satisfaction_level : num 0.38 0.8 0.11 0.72 0.37 0.41 0.1 0.92 0.89 0.42 ...

$ last_evaluation : num 0.53 0.86 0.88 0.87 0.52 0.5 0.77 0.85 1 0.53 ...

$ number_project : int 2 5 7 5 2 2 6 5 5 2 ...

$ average_montly_hours : int 157 262 272 223 159 153 247 259 224 142 ...

$ time_spend_company : int 3 6 4 5 3 3 4 5 5 3 ...

$ Work_accident : int 0 0 0 0 0 0 0 0 0 0 ...

$ left : int 1 1 1 1 1 1 1 1 1 1 ...

$ promotion_last_5years: int 0 0 0 0 0 0 0 0 0 0 ...

$ Department : Factor w/