Loading [a11y]/accessibility-menu.js
Predicting Employee Attrition using Machine Learning | IEEE Conference Publication | IEEE Xplore

Predicting Employee Attrition using Machine Learning


Abstract:

The growing interest in machine learning among business leaders and decision makers demands that researchers explore its use within business organisations. One of the maj...Show More

Abstract:

The growing interest in machine learning among business leaders and decision makers demands that researchers explore its use within business organisations. One of the major issues facing business leaders within companies is the loss of talented employees. This research studies employee attrition using machine learning models. Using a synthetic data created by IBM Watson, three main experiments were conducted to predict employee attrition. The first experiment involved training the original class-imbalanced dataset with the following machine learning models: support vector machine (SVM) with several kernel functions, random forest and K-nearest neighbour (KNN). The second experiment focused on using adaptive synthetic (ADASYN) approach to overcome class imbalance, then retraining on the new dataset using the abovementioned machine learning models. The third experiment involved using manual undersampling of the data to balance between classes. As a result, training an ADASYN-balanced dataset with KNN (K = 3) achieved the highest performance, with 0.93 F1-score. Finally, by using feature selection and random forest, F1-score of 0.909 was achieved using 12 features out of a total of 29 features.
Date of Conference: 18-19 November 2018
Date Added to IEEE Xplore: 10 January 2019
ISBN Information:
Print on Demand(PoD) ISSN: 2325-5498
Conference Location: Al Ain, United Arab Emirates

References

References is not available for this document.