AI-Based Employee Attrition Prediction System
Python, Flask, Machine Learning, Scikit-learn, Pandas, NumPy, Joblib, HTML, CSS
About this project
AI-Based Employee Attrition Prediction System is a Machine Learning web application designed to predict whether an employee is likely to leave an organization. The system analyzes employee-related factors such as age, monthly income, job satisfaction, years at company, overtime, work-life balance, and total working years. A trained Machine Learning classification model provides an attrition prediction along with a probability/confidence score.
What this project does
- Employee attrition prediction using Machine Learning
- Predicts Likely to Leave / Likely to Stay
- Attrition probability percentage
- Employee information input form
- Job satisfaction analysis
- Monthly income consideration
- Overtime analysis
- Work-life balance analysis
- Experience and years-at-company analysis
- Machine Learning classification model
- Simple Flask web application
- User-friendly HTML/CSS interface
- Dataset-based model training
- Beginner-friendly project structure
Requirements
Python 3.9 or above
Flask
Pandas
NumPy
Scikit-learn
Joblib
VS Code or any Python IDE
Web browser
How to run the project
1. Extract the project ZIP file.
2. Open the project folder in VS Code.
3. Open Terminal.
4. Create virtual environment:
python -m venv venv
5. Activate virtual environment:
venv\Scripts\activate
6. Install required packages:
pip install -r requirements.txt
7. Train the Machine Learning model:
python train_model.py
8. Run the Flask application:
python app.py
9. Open in browser:
http://127.0.0.1:5000/
Make it your own
Explore the implementation, understand the decisions and adapt the project to your requirements.
Files & downloads
Choose only the resources you need.
Paid downloads stay available in your account when you return to this page. Free resources are supported by ads.