R语言-向量机-员工离职预测训练赛

最新推荐文章于 2020-12-23 10:10:55 发布

拦羊嗓子回牛声

最新推荐文章于 2020-12-23 10:10:55 发布

阅读量1.2k

点赞数

分类专栏： R语言代码文章标签： r语言练习

本文链接：https://blog.csdn.net/wlt9037/article/details/77930949

版权

本文介绍了一个基于R语言的员工离职预测训练赛，数据包含影响离职的多种因素，如工资、出差、满意度等。训练集有1100条记录，测试集有350条，缺失离职状态。目标是使用训练数据建立模型，预测测试数据的离职情况。

摘要由CSDN通过智能技术生成

题目：员工离职预测训练赛
网址：http://www.pkbigdata.com/common/cmpt/员工离职预测训练赛_竞赛信息.html
要求：
数据主要包括影响员工离职的各种因素（工资、出差、工作环境满意度、工作投入度、是否加班、是否升职、工资提升比例等）以及员工是否已经离职的对应记录。
数据分为训练数据和测试数据，分别保存在pfm_train.csv和pfm_test.csv两个文件中。
其中训练数据主要包括1100条记录，31个字段。
测试数据主要包括350条记录，30个字段，跟训练数据的不同是测试数据并不包括员工是否已经离职的记录，学员需要通过由训练数据所建立的模型以及所给的测试数据，得出测试数据相应的员工是否已经离职的预测。
数据：https://pan.baidu.com/s/1qXZOS8W 密码：bxgm

代码：

#######################  训练  #############################
data <- read.csv("E:/.../pfm_train.csv", sep=",", header=TRUE)
colnames(data)[1]<-c("Age")     #首列列名乱码
data[,2] <- as.factor(as.vector(data)[,2])

library(e1071)
#########线性支持向量机
#########传递给函数svm()的关键参数是kernel、cost和gamma。Kernel指的是支持向量机的类型，它可能是线性SVM、多项式SVM、径向SVM或Sigmoid SVM。Cost是违反约束时的成本函数，gamma是除线性SVM外其余所有SVM都使