Skip to content

vyawaha/employee-performance-predictor

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

🧠 Employee Performance Predictor using Machine Learning

Python Machine Learning Status Scikit-learn


📌 Project Overview

The Employee Performance Predictor is a Machine Learning-based HR analytics system that predicts employee performance levels:

  • 🟢 High Performer
  • 🟡 Medium Performer
  • 🔴 Low Performer

It uses structured HR data such as experience, salary, attendance, training hours, and feedback scores to generate predictions.


🎯 Business Problem

Organizations struggle with:

  • Subjective employee evaluation
  • Delayed performance feedback
  • Inefficient training allocation
  • Bias in promotions

This system solves it using data-driven HR decision-making.


⚙️ Tech Stack

  • Python
  • Pandas, NumPy
  • Scikit-learn
  • Random Forest Classifier
  • Joblib (Model Saving)

🧠 Machine Learning Workflow

Data Generation → Feature Engineering → Model Training → Evaluation → Prediction → HR Recommendation


📊 Features Used

  • Age
  • Experience
  • Salary
  • Training Hours
  • Attendance Rate
  • Projects Completed
  • Feedback Score
  • Department

🚀 How It Works

  1. Synthetic HR dataset is generated
  2. Model is trained using Random Forest
  3. Employee data is given as input
  4. System predicts performance level
  5. HR recommendation is generated

💻 How to Run

pip install -r requirements.txt
python main.py

📸 Sample Output
👤 Employee: Rahul Sharma
🎯 Predicted Performance: High
📌 HR Recommendation: Promotion Recommended

🏗️ Project Structure
Employee-Performance-Predictor/
│
├── src/
│   ├── data_generator.py
│   ├── train_model.py
│   ├── predict.py
│
├── data/
├── models/
├── images/
├── main.py
└── README.md

📈 Business Impact
Improves HR decision accuracy
Reduces bias in performance reviews
Identifies training needs early
Helps in promotion planning

🔮 Future Enhancements
Streamlit Dashboard UI
SHAP Explainability
Real HR dataset integration
Cloud deployment (AWS/Render)
Email alert system for low performers

👨‍💻 Author
Muktai Vyawahare
Computer Science Engineering Student
AI/ML Developer

About

ML-based HR system predicting employee performance using Random Forest

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages