NU-JST
Vol 1 No 2 (2024): Published in November of 2024

Exploratory Data Analysis dan Machine Learning dalam Memprediksi Employee Attrition: Exploratory Data Analysis and Machine Learning in Predicting Employee Attrition

Kholifah, Binti (Unknown)
Firmansyah, Fendy Bayu (Unknown)
Sururi, Nafis (Unknown)
Nugraha, Danang Satya (Unknown)



Article Info

Publish Date
30 Nov 2024

Abstract

Employee attrition is one of the main challenges faced by organizations in retaining competent human resources. This study aims to explore data patterns and predict employee attrition using the Exploratory Data Analysis (EDA) approach and Machine Learning algorithms such as Logistic Regression, Support Vector Machine (SVM), and Naive Bayes. The analysis was conducted on a dataset that includes various factors such as demographics, job satisfaction, and employee performance. The research findings indicate that Logistic Regression achieved an accuracy of 87%, but the model has weaknesses in detecting the positive class (attrition), as reflected by its low recall score. SVM, with an accuracy of 85%, provided high precision for the positive class but performed poorly in detecting actual attrition cases. Conversely, Naive Bayes, with an accuracy of 85%, demonstrated more balanced performance with a higher weighted average F1-score compared to the other models, although there is still room for improvement, particularly in predicting the positive class. Based on the results, Naive Bayes stands out as a more reliable model for predicting employee attrition with more balanced performance compared to Logistic Regression and SVM. To enhance prediction performance, it is recommended to address the class imbalance in the dataset through techniques such as oversampling, undersampling, class weighting, or specialized algorithms.

Copyrights © 2024






Journal Info

Abbrev

nujst

Publisher

Subject

Chemistry Civil Engineering, Building, Construction & Architecture Computer Science & IT Engineering Materials Science & Nanotechnology

Description

NU-JST is a national scale journal that covers various issues and studies in the fields of science and technology. The aim of this educational journal is to disseminate conceptual thoughts and research results that have been achieved in the fields of Pharmacy, Mathematics and Natural Sciences, ...