matlab 的流量代码,用在人力资源预测上其实还挺有意思的。项目是基于 Kaggle 上的一个员工数据集,字段挺全的,比如员工满意度
、平均每月工时
这些,拿来练模型方便。
香草神经网络、支持向量机这些经典模型都试了一遍,还搭了个带反向传播的多层感知器
,运行在 Matlab 里,代码不复杂,逻辑也清晰。
数据分成了两块:原始的放在原始 Kaggle 数据集
里,好的就在清除的数据
文件夹里。清洗过程可以学不少技巧,是怎么缺失值和分类字段。
重点是,嗯……它计划后续用Keras/TensorFlow重构一版——对你要迁移到深度学习框架的项目来说,也是一种思路参考。
如果你最近也在琢磨员工离职预测这种场景,这套代码挺值得翻一翻的。运行效果还不错,尤其适合做些初步验证和调试。
哦对了,还有些相关文章也蛮有用,像神经网络 MATLAB 程序、神经网络案例源码,可以顺手参考一下。