Skip to content

Yashvj22/Employee_Attrition_Classification_Model

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

4 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

Employee Attrition Prediction

This project predicts employee attrition based on various workplace and personal factors. The model is built using Machine Learning techniques and deployed as an interactive Streamlit app.

πŸš€ Features

  • Exploratory Data Analysis (EDA): Visualizing key factors influencing employee attrition.
  • Feature Engineering: Created meaningful features using Best-fit Random Forest.
  • Model Building: Implemented Voting Classifier, Bagging Classifier, and Random Forest.
  • Hyperparameter Tuning: Used Optuna to optimize model performance.
  • Employee Attrition Prediction App: Users can input details and get real-time predictions.

🎯 How to Use

  1. Open the Streamlit App (link below).
  2. Enter the required input details about an employee.
  3. Click Predict Attrition to see whether the employee is likely to stay or leave.

πŸ“‚ Dataset

The dataset contains key employee-related features, including:

  • Age, Monthly Income, Distance from Home, Education Level, Number of Dependents, Company Tenure
  • Remote Work, Leadership Opportunities, Company Reputation, etc.

The target variable is Attrition Status (Stay or Leave).

πŸ”— Live Demo

πŸš€ Check out the live app here: Employee Attrition Prediction

About

This project focuses on predicting employee attrition using machine learning models. The goal is to analyze key factors influencing attrition and build an accurate predictive model to help organizations retain valuable employees.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors