E. Dritsas, M. Trigka, Ph. Mylonas |
Comparative Analysis of Machine Learning Models for Employee Salary Prediction |
5th International Conference on Novel & Intelligent Digital Systems (NIDS 2025), September 24-26, 2025, Athens, Greece |
ABSTRACT
|
This study investigates the application of seven supervised machine learning (ML) models for predicting employee salaries based on demographic, educational, and occupational attributes. The dataset comprises both continuous and high-cardinality categorical features, for which suitable encoding was applied to facilitate model training. Models evaluated include Linear and Ridge Regression (LinR, RidgeR), Decision Trees (DT), K-Nearest Neighbors (KNN), Support Vector Regression (SVR), Gradient Boosting (GB), and Random Forest (RF). Performance is measured using Root Mean Squared Error (RMSE), Mean Absolute Error (MAE), and R-squared (R2) metrics. Results show that ensemble models, particularly RF and GB, achieve the highest accuracy, with RF reaching an R2 of 0.94. In contrast, kernel and instance-based methods underperform due to limitations in handling categorical data. The findings support the integration of ensemble models into human resource (HR) analytics systems and highlight the importance of model selection based on data structure.
|
24 September, 2025 |
E. Dritsas, M. Trigka, Ph. Mylonas, "Comparative Analysis of Machine Learning Models for Employee Salary Prediction", 5th International Conference on Novel & Intelligent Digital Systems (NIDS 2025), September 24-26, 2025, Athens, Greece |
[ PDF] [
BibTex] [
Print] [
Back] |