Skip to content

The aim is to build a predictive model that can accurately classify whether the employee is likely to leave or the employee is likely to stay in the company. This allows companies to take proactive measures, such as improving working conditions, offering promotions, or addressing dissatisfaction, to retain valuable employees.

Notifications You must be signed in to change notification settings

nirajan-jha/Predicting-Employee-Churn-Using-Machine-Learning

Repository files navigation

Predicting-Employee-Churn-Using-Machine-Learning

Overview

The aim is to build a predictive model that can accurately classify whether the employee is likely to leave or the employee is likely to stay in the company. This allows companies to take proactive measures, such as improving working conditions, offering promotions, or addressing dissatisfaction, to retain valuable employees.

Aim:

Based on different factors, a model is created that can classify whether the employee will leave the company or not. It is a supervised classification problem statement.

About the Dataset

The dataset contains information about the employee leaving the company. I have recorded 14999 employees from the HR for the analysis purpose based on different features/factors that include 9 independent features and 1 output feature i.e. target variable.

Brief Information About Each Columns of the Dataset

  1. Satisfaction level: Employee satisfaction level (ranges from 0 to 1).
  2. Last evaluation: Last evaluation score of the employee (ranges from 0 to 1).
  3. Number of project: Number of projects the employee has worked on.
  4. Average monthly hours: Average monthly hours the employee has worked.
  5. Time spend company: Number of years the employee has spent in the company.
  6. Work accident: Whether the employee has had a work accident (0 for No, 1 for Yes).
  7. Left: Whether the employee has left the company (0 for No, 1 for Yes). This is the target variable we aim to predict using the other features.
  8. Promotion last 5 years: Whether the employee was promoted in the last 5 years (0 for No, 1 for Yes).
  9. Departments: Department in which the employee works. 10 Salary: Salary level of the employee (low, medium, high).

Implemented Machine Learning Algorithm:

  1. Logistic Regression
  2. Decision Tree
  3. Random Forest

About

The aim is to build a predictive model that can accurately classify whether the employee is likely to leave or the employee is likely to stay in the company. This allows companies to take proactive measures, such as improving working conditions, offering promotions, or addressing dissatisfaction, to retain valuable employees.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published