← Project collection
Mini project · Semester 5 · Computer Science and Engineering

AI-Based Employee Attrition Prediction System

Python, Flask, Machine Learning, Scikit-learn, Pandas, NumPy, Joblib, HTML, CSS

4 visitors0 file downloads0 screenshots
THE IDEA

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.

YOUR PROJECT TOOLKIT

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.