Using machine learning algorithms with improved accuracy to analyze and predict employee attrition

被引:1
|
作者
Alsubaie, Fiyhan [1 ]
Aldoukhi, Murtadha [2 ]
机构
[1] Natl Commiss Acad Accreditat & Assessment, Riyadh 13318, Saudi Arabia
[2] Alyamamah Univ, Coll Engn & Architecture, Ind Engn Dept, Riyadh 11512, Saudi Arabia
关键词
Machine Learning; Employee Attrition; Improve Model Accuracy; Prediction; Decision Tree; Random Forest; Binary Logistic Regression;
D O I
10.5267/dsl.2023.12.006
中图分类号
C93 [管理学]; O22 [运筹学];
学科分类号
070105 ; 12 ; 1201 ; 1202 ; 120202 ;
摘要
Human migration is based on pull factors that individuals evaluate when it comes to moving to a different territory. Likewise, employee attrition is a phenomenon that represents the tendency to a reduction in employees within an organization. This research paper aims to develop and evaluate machine learning algorithms, namely Decision Tree, Random Forest, and Binary Logistic Regression, to predict employee attrition using the IBM dataset available on Kaggle. The objective is to provide organizations with a proactive approach to employee retention and human resource management by creating accurate predictive models. Employee attrition has significant implications for an organization's reputation, profitability, and overall structure. By accurately predicting employee attrition, organizations can identify the factors contributing to it and implement data-driven human resources management practices. This study contributes to improving decision-making processes, including hiring and firing decisions, and ultimately enhances an organization's capital. The IBM dataset used in this study consists of anonymized employee records and their employment outcomes. It provides a comprehensive HR data representation for analysis and prediction. Three machine learning algorithms, Decision Tree, Random Forest, and Binary Logistic Regression, were utilized in this research. These algorithms were selected for their potential to improve accuracy in predicting employee attrition. The Logistic Regression model yielded the highest accuracy of 87.44% among the tested algorithms. By leveraging this study's findings, organizations can develop predictive models to identify factors contributing to employee attrition. These insights can inform strategic decisions and optimize human resource management practices.
引用
收藏
页码:1 / 18
页数:18
相关论文
共 50 条
  • [21] Employee attrition prediction in a pharmaceutical company using both machine learning approach and qualitative data
    Mozaffari, Fatemeh
    Rahimi, Marzieh
    Yazdani, Hamidreza
    Sohrabi, Babak
    BENCHMARKING-AN INTERNATIONAL JOURNAL, 2023, 30 (10) : 4140 - 4173
  • [22] Using machine learning algorithms to predict groundwater levels in Indonesian
    Hikouei, Iman Salehi
    Eshleman, Keith N.
    Saharjo, Bambang Hero
    Graham, Laura L. B.
    Applegate, Grahame
    Cochrane, Mark A.
    SCIENCE OF THE TOTAL ENVIRONMENT, 2023, 857
  • [23] Comparing the performance of machine learning algorithms using estimated accuracy
    Gupta S.
    Saluja K.
    Goyal A.
    Vajpayee A.
    Tiwari V.
    Measurement: Sensors, 2022, 24
  • [24] Machine Learning Based Predictive Model for Risk Assessment of Employee Attrition
    Gabrani, Goldie
    Kwatra, Anshul
    COMPUTATIONAL SCIENCE AND ITS APPLICATIONS - ICCSA 2018, PT IV, 2018, 10963 : 189 - 201
  • [25] Improved NO2 Prediction Using Machine Learning Algorithms
    Jaja-Wachuku, Chukwuemeka
    Garbagna, Lorenzo
    Saheer, Lakshmi Babu
    Oghaz, Mandi Maktab Dar
    ARTIFICIAL INTELLIGENCE APPLICATIONS AND INNOVATIONS, PT II, AIAI 2024, 2024, 712 : 215 - 225
  • [26] Using Machine Learning to Predict Effective Compression Algorithms for Heterogeneous Datasets
    Burtchell, Brandon Alexander
    Burtscher, Martin
    2024 DATA COMPRESSION CONFERENCE, DCC, 2024, : 183 - 192
  • [27] Predict Hyperkalemia in Advanced CKD Patients Using Machine Learning Algorithms
    Chang, Hsin-Hsiung
    Wu, Chia-Lin
    JOURNAL OF THE AMERICAN SOCIETY OF NEPHROLOGY, 2021, 32 (10): : 727 - 727
  • [28] Using Machine Learning Algorithms to Predict the State of Financial Inclusion in Africa
    Ismail, Qusai F.
    Al-Sobh, Eslam S.
    Al-Omari, Sarah S.
    Yaseen, Tuqa M. Bani
    Abdullah, Malak A.
    2021 12TH INTERNATIONAL CONFERENCE ON INFORMATION AND COMMUNICATION SYSTEMS (ICICS), 2021, : 317 - 323
  • [29] Accuracy detection of coronary artery disease using machine learning algorithms
    Singh, Harinder
    Rehman, Tasneem Bano
    Gangadhar, Ch
    Anand, Rohit
    Sindhwani, Nidhi
    Babu, M. Vijaya Sekhar
    APPLIED NANOSCIENCE, 2021, 13 (3) : 1791 - 1791
  • [30] Employee Attrition Prediction using Nested Ensemble Learning Techniques
    Alshiddy, Muneera Saad
    Aljaber, Bader Nasser
    INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2023, 14 (07) : 932 - 938