Employee attrition prediction using machine learning models (#498)

Authors

Iparraguirre-Villanueva, Orlando

Chauca-Huete, Luis

Prieto-Chavez, Rosas

Paulino-Moreno, Cleoge

Abstract

Today's business landscape is characterized by competition and dynamism, which has transformed human resource management into an essential strategic partner for organizations. Employee turnover poses risks that affect productivity and knowledge management. This study focuses on predicting employee turnover using Machine Learning (ML) models. For the training process, a dataset composed of 4410 records and 29 variables was used, in the process of training and evaluation of the ten models, the artificial intelligence (AI) method was followed. The findings showed that the XG Boost Classifier (XGBC) and Random Forest (RF) models achieved the best accuracy and performance rates, with 98.8% and 98.7%. Followed by Decision Tree Classifier (DT) with 97.6%, and the other models, such as Gradient Boosting Classifier (GBC), Ada boost Classifier (AC), Logistic Regression (LR), KN Classifier (KNNC), SGD Classifier (SGDC), Support Vector Classifier (SVC) and Nu Support Vector Classifier (NuSVC), achieved the following rates: 88.4%, 85.4%, 84%, 82.2%, 83.0%, 83.0%, 55.0%, respectively. Finally, it is concluded that the models are useful and effective in prediction. Their practical implementation in human resource management strategies is recommended for proactive intervention.