Paper
28 February 2023 Employee turnover prediction using machine learning models
Chenyu Liao
Author Affiliations +
Proceedings Volume 12596, International Conference on Mechatronics Engineering and Artificial Intelligence (MEAI 2022); 125960X (2023) https://doi.org/10.1117/12.2672733
Event: International Conference on Mechatronics Engineering and Artificial Intelligence (MEAI 2022), 2022, Changsha, China
Abstract
With respect to the 16 characteristics of the workers, the objective of this study is to investigate how employee turnover can be classified using various machine learning algorithms (Support Vector Classification, Decision Tree Classifier, AdaBoost Classifier, Random Forest Classifier, Extra Trees Classifier, Logistic Regression and Gradient Boosting Classifiers). The information comes from the Employee Turnover dataset by E. Babushkin. Seven distinct classification models were developed and contrasted, including naive Bayes, random forest, logistic regression, support vector machines, and XGBoost. Numerous experiments validate the effectiveness of machine learning model. Among all the models, we find that the random forest model achieves the best results, which can be furtherly utilized in real-world prediction.
© (2023) COPYRIGHT Society of Photo-Optical Instrumentation Engineers (SPIE). Downloading of the abstract is permitted for personal use only.
Chenyu Liao "Employee turnover prediction using machine learning models", Proc. SPIE 12596, International Conference on Mechatronics Engineering and Artificial Intelligence (MEAI 2022), 125960X (28 February 2023); https://doi.org/10.1117/12.2672733
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Machine learning

Decision trees

Random forests

Industry

Data modeling

Visualization

Data visualization

Back to Top