Enhancing Employee Attrition Prediction Using Hybrid Machine Learning Models

Project Code :TCMAPY2287

Objective

The project aims to develop a robust, production-ready employee attrition prediction system using tree-based and ensemble machine learning techniques. Key objectives include implementing and comparing classifiers such as Stacking Classifier, Voting Classifier, XGBoost, Decision Tree, and Random Forest, with evaluation via accuracy, precision, recall, and macro-averaged F1-score to address class imbalance—where Random Forest emerges as the top performer. A secure Flask-based web application is built to enable user registration/login (stored in MySQL), CSV upload with data preview and feature validation, interactive single-record predictions with model selection, and display of pre-computed performance metrics. Secure credential handling and prediction audit logs are maintained in MySQL, delivering an intuitive, no-code interface for HR professionals to forecast and mitigate attrition risks effectively.

Abstract

Employee attrition prediction is crucial for organizations to mitigate the risk of losing top talent, ensuring smooth operations and reducing costs associated with recruitment and training. This project aims to enhance employee attrition prediction by utilizing hybrid machine learning models, combining both classical and advanced algorithms for better accuracy. The system integrates a Flask web application with a MySQL database for managing employee records, model training, and predictions. The system employs multiple machine learning models, including Stacking Classifier, Voting Classifier, XGBoost, Decision Tree, and Random Forest, to classify whether an employee is likely to leave the organization. Each model's performance metrics, including accuracy, precision, recall, and F1-score, are computed to evaluate and select the best-performing model. The system allows users to upload employee data in CSV format, and the prediction is made based on selected models. The web interface is user-friendly, providing clear feedback to users and making predictions accessible. The integration of machine learning models with real-world data through a web interface provides organizations with valuable insights into employee retention, helping HR departments to take proactive actions and reduce turnover.

Keywords: Employee Attrition, Machine Learning, Flask, MySQL, Hybrid Models, Prediction, Random Forest, XGBoost, Stacking Classifier, Voting Classifier, Decision Tree, Web Application.

NOTE: Without the concern of our team, please don't submit to the college. This Abstract varies based on student requirements.

Block Diagram

Specifications

SOFTWARE REQUIREMENS

Operating System                               :  Windows 7/8/10

Server side Script                                :  Flask

Programming Language                     :  Python

Libraries                                              : Flask, Tensorflow, Pandas, Torch, Keras, Sklearn,Numpy , Seaborn

IDE/Workbench                                  :  VSCode

Server Deployment                             :  Xampp Server

Database                                             :  mysql

HARDWARE REQUIREMENTS

Processor                                   - I3/Intel Processor

RAM                                       - 8GB (min)

Hard Disk                                - 128 GB

Key Board                               - Standard Windows Keyboard

Mouse                                      - Two or Three Button Mouse

Monitor                                    - Any

Demo Video

mail-banner
call-banner
contact-banner
Request Video