Predictive Modelling for Employee Attrition: A Comparative Study of Machine Learning Algorithms
摘要
Predicting workforce turnover in the dynamic tech industry is vital for employers and job seekers. Employee churn rate is a major concern for corporation, affecting both productivity and operational continuity. Traditional methods of predicting employee turnover often fall short in accuracy and efficiency. This study explores the utilization of machine learning techniques to forecast employee attrition, comparing four algorithms: Logistic Regression, Tree-based decision models, Random Forest Algorithm, and Gradient Boosting Techniques. Data from a corporate HR dataset was pre-processed and split into training and testing sets, with categorical variables encoded and features scaled. Logistic Regression achieved a validity of 89.12%, precision of 68.42%, recall of 33.33%, and F1 score of 44.83%. The Gradient Boosting Classifier followed closely with an accuracy of 88.78%, precision of 65.00%, recall of 33.33%, and F1 score of 44.07%. The Random Forest and Decision Tree classifiers showed lower performance metrics. The results demonstrate that leveraging innovative machine learning models can markedly boost the accuracy of attrition prediction, aiding HR departments in proactive employee retention strategies. This research highlights the potential for further refinement and application in various organizational contexts.