AI-Based Employee Attrition Prediction System

Before you download

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.

Getting started

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/

Requirements

Python 3.9 or above
Flask
Pandas
NumPy
Scikit-learn
Joblib
VS Code or any Python IDE
Web browser

Build your understanding

Use this resource as a learning reference. Explore how the code works, test it with your own inputs and document your changes for your presentation.