Predict employee attrition using machine learning and gain actionable insights through data visualization.
This project helps HR departments predict which employees are likely to leave the organization and understand the key reasons behind attrition. By analyzing employee data and visualizing trends, companies can take timely action to improve retention.
- Hansa Pradhan
- Snehal Nivsarkar
- Sujata Rai
- Gursimran Singh
Supervised by: Dr. Jabir Ali, Assistant Professor
SRM Institute of Science & Technology
Employee attrition is a costly and disruptive problem. Most organizations lack a data-driven approach to predict and mitigate turnover. This project leverages machine learning to forecast attrition risk and identify contributing factors.
- Predict probability of employee attrition using Random Forest
- Visualize important factors driving attrition
- Enable HR to take preventive actions
- Build a user-friendly interface for secure HR access
-
Data Preprocessing
- Cleaned and prepared HR dataset (~1700 records)
- Created target labels for attrition
-
Modeling
- Used Random Forest Classifier for prediction
- Tuned hyperparameters for optimal accuracy
- Analyzed feature importance
-
Visualization
- Dashboards showing attrition risk and key metrics
- Attrition List flagging high-risk employees
-
Interface
- Login page for HR access
- Navigation through dashboard, predictions, and survey forms
- Language: Python
- Algorithm: Random Forest
- Visualization: Matplotlib, Seaborn
- IDE: Jupyter Notebook / Any Python IDE
- Libraries:
pandas,scikit-learn,matplotlib,seaborn