使用深度学习模型预测员工离职

更新预告:使用3个深度学习模型完成对Kaggle公开数据集"HR Analytics Case Study"的分类预测(目前文章更新中)。

本文旨在探讨深度学习模型简单运用到表格数据的学习上的可行性,使用数据集来至于kaggle。

 总览

        首先对数据集进行预处理,处理后分别使用MLP、1DCNN、TabNet模型来学习并预测,引入Accuracy、Precision、Recall、F1-Score来评估模型。最后结合模型的输出结果来简单分析影响员工离职的关键因数。

数据处理

数据说明

        该数据集包含5个表共计29个变量、4410个样本,其中21个数字型变量,8个字符型变量,其中有两个表记录的是2015年全年的员工上下班打卡记录,节假日与周末除外。Attrition为预测变量。

数据预处理

无关变量去除

        去除Over18、StandarHours、EmployeeCount、EmployeeNumber这些常量(固定值)

持续更新中......

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值