Employee Attrition Prediction Using Machine Learning Techniques

Authors

DOI:

https://doi.org/10.31642/JoKMC/2018/110211

Keywords:

Naive Bayes, machine learning, SVM, KNN, Employ Attrition

Abstract

Effective employees are regarded as the most precious assets and the foundation of any business. Companies invest a lot of money in staff training programs because they believe they will pay off in the long run. Therefore, it is crucial to keep a long-term, promising staff; as has been seen throughout the years, this is one of HR's most challenging duties. This study aims to determine the key variables influencing employee attrition and develop a machine learning model to predict employee attrition based on the variables given. This will make it easier for management to spot workers who are likely to quit, allowing them to take preventative measures and make wise decisions about appraisal and recognition. The four most popular classification algorithms utilized in this study were KNN, Naive Bayes, Random Forest and Logistic Regression. It was shown that Naive Bayes classifier outperformed the others by 89% in terms of accuracy and produced more accurate predictions.

Downloads

Download data is not yet available.

References

S. S. Alduayj and K. Rajpoot, “Predicting employee attrition using machine learning,” in 2018 international conference on innovations in information technology (iit), 2018, pp. 93–98. DOI: https://doi.org/10.1109/INNOVATIONS.2018.8605976

F. Fallucchi, M. Coladangelo, R. Giuliano, and E. William De Luca, “Predicting employee attrition using machine learning techniques,” Computers, vol. 9, no. 4, p. 86, 2020. DOI: https://doi.org/10.3390/computers9040086

A. Raza, K. Munir, M. Almutairi, F. Younas, and M. M. S. Fareed, “Predicting employee attrition using machine learning approaches,” Appl. Sci., vol. 12, no. 13, p. 6424, 2022. DOI: https://doi.org/10.3390/app12136424

P. Ajit, “Prediction of employee turnover in organizations using machine learning algorithms,” Algorithms, vol. 4, no. 5, p. C5, 2016.

M. Subhashini and R. Gopinath, “Employee attrition prediction in industry using machine learning techniques,” Int. J. Adv. Res. Eng. Technol., vol. 11, no. 12, pp. 3329–3341, 2020.

A. Qutub, A. Al-Mehmadi, M. Al-Hssan, R. Aljohani, and H. S. Alghamdi, “Prediction of employee attrition using machine learning and ensemble methods,” Int. J. Mach. Learn. Comput, vol. 11, no. 2, pp. 110–114, 2021. DOI: https://doi.org/10.18178/ijmlc.2021.11.2.1022

F. K. Alsheref, I. E. Fattoh, and W. M Ead, “Automated Prediction of Employee Attrition Using Ensemble Model Based on Machine Learning Algorithms,” Comput. Intell. Neurosci., vol. 2022, 2022. DOI: https://doi.org/10.1155/2022/7728668

S. Najafi-Zangeneh, N. Shams-Gharneh, A. Arjomandi-Nezhad, and S. Hashemkhani Zolfani, “An improved machine learning-based employees attrition prediction framework with emphasis on feature selection,” Mathematics, vol. 9, no. 11, p. 1226, 2021. DOI: https://doi.org/10.3390/math9111226

P. K. Jain, M. Jain, and R. Pamula, “Explaining and predicting employees’ attrition: a machine learning approach,” SN Appl. Sci., vol. 2, pp. 1–11, 2020. DOI: https://doi.org/10.1007/s42452-020-2519-4

V. V. Saradhi and G. K. Palshikar, “Employee churn prediction,” Expert Syst. Appl., vol. 38, no. 3, pp. 1999–2006, 2011. DOI: https://doi.org/10.1016/j.eswa.2010.07.134

C. Singh and A. Gopal, “Per for mance Analysis of Faculty using Data Mining Techniques,” Int. J. Comput. Sci. Appl. Issue, pp. 140–144, 2010.

R. Punnoose and P. Ajit, “Prediction of Employee Turnover in Organizations using

Machine Learning Algorithms,” Int. J. Adv. Conference on Contemporary Computing and Informatics (IC3I) (Vol. 6, pp. 2361-2367).

IEEE.

Res. Artif. Intell., vol. 5, no. 9, pp. 22–26, [22] Javatpoint, “K-Nearest Neighbor(KNN)

, doi: 10.14569/ijarai.2016.050904. Algorithm for Machine Learning - DOI: https://doi.org/10.14569/IJARAI.2016.050904

R. Chakraborty, K. Mridha, R. N. Shaw, and Javatpoint.”

A. Ghosh, “Study and prediction analysis of https://www.javatpoint.com/k-nearest-neighb

the employee turnover using machine or-algorithm-for-machine-learning (accessed

learning approaches,” in 2021 IEEE 4th Oct. 05, 2023).

International Conference on Computing, [23] Iyortsuun, N. K., Kim, S. H., Jhon, M., Yang,

Power and Communication Technologies H. J., & Pant, S. (2023, January). A review of

(GUCON), 2021, pp. 1–6.

S. Al-Darraji, D. G. Honi, F. Fallucchi, A. I. Abdulsada, R. Giuliano, and H. A. Abdulmalik, “Employee attrition prediction using deep neural networks,” Computers, vol. 10, no. 11, pp. 1–11, 2021, doi: 10.3390/computers10110141. DOI: https://doi.org/10.3390/computers10110141

S. M. Arqawi et al., “Predicting Employee Attrition and Performance Using Deep Learning,” J. Theor. Appl. Inf. Technol., vol. 100, no. 21, pp. 6526–6536, 2022.

K. K. Mohbey, “Employee’s attrition prediction using machine learning approaches,” in Machine Learning and Deep Learning in Real-Time Applications, IGI Global, 2020, pp. 121–128. DOI: https://doi.org/10.4018/978-1-7998-3095-5.ch005

I. F. Alsuahim, F. A. Alotaibi, M. A. AlAsiri,

M. S. Alkharji, and S. A. Alharthi, “Predicting employee attrition using machine learning,” Proc. Int. Conf. Ind. Eng. Oper. Manag., no. November, pp. 1007–1014, 2019.

R. Yedida, R. Reddy, R. Vahi, R. Jana, A. GV, and D. Kulkarni, “Employee Attrition Prediction,” 2018, [Online]. Available: http://arxiv.org/abs/1806.10480

S. Aggarwal, M. Singh, S. Chauhan, M. Sharma, and D. Jain, “Employee attrition prediction using machine learning comparative study,” in Intelligent Manufacturing and Energy Sustainability: Proceedings of ICIMES 2021, 2022, pp. 453–466. DOI: https://doi.org/10.1007/978-981-16-6482-3_45

L. S. Ganthi, Y. Nallapaneni, D. Perumalsamy, and K. Mahalingam, “Employee Attrition Prediction Using Machine Learning Algorithms,” in Proceedings of International Conference on Data Science and Applications: ICDSA 2021, Volume 1, 2022, pp. 577–596. DOI: https://doi.org/10.1007/978-981-16-5120-5_44

Rana, A., Malik, S., Chauhan, M., & Kumar, P. (2023, September). Heart Disease Predictive Analysis Using Machine Learning Approaches. In 2023 6th International DOI: https://doi.org/10.1109/IC3I59117.2023.10397609

machine learning and deep learning approaches on mental health diagnosis. In Healthcare (Vol. 11, No. 3, p. 285). MDPI. DOI: https://doi.org/10.3390/healthcare11030285

Downloads

Published

2025-05-19

How to Cite

Dr. Anamika Rana, Sushma Malik, & madhu Chauhan. (2025). Employee Attrition Prediction Using Machine Learning Techniques. Journal of Kufa for Mathematics and Computer, 11(2), 88-100. https://doi.org/10.31642/JoKMC/2018/110211

Share