Giter Club home page Giter Club logo

thanakorntha / salifort-employee-churn-prediction Goto Github PK

View Code? Open in Web Editor NEW
0.0 1.0 0.0 11.01 MB

This project is a capstone part of the Google Advanced Data Analytics Professional Certificate on Coursera. This project involves data preparation and cleaning, exploratory data analysis (EDA), feature engineering, and model building and evaluation. Machine learning techniques are Logistic Regression, Decision Tree, Random Forest and XGBoost.

License: MIT License

Jupyter Notebook 100.00%
data-analysis data-cleaning data-visualization decision-trees employee-retention logistic-regression modeling python random-forest xgboost

salifort-employee-churn-prediction's Introduction

Salifort Motors: Predicting Employee Churn

Overview

This project aimed to develop the best machine learning model, among four models (Logistic Regression, Decision Tree, Random Forest, and XGBoost), to predict whether or not an employee will leave the company. It utilized employee data collected from the HR department at the company. The final random forest model achieved a 91% recall and 92% precision in identifying the most important factors influencing employee churn. The model identified the number of projects, average monthly hours, the last evaluation score, and time spent at the company as the most significant factors differentiating leavers from non-leavers.

Business Understanding

Salifort Motors, a leading global manufacturer of alternative energy vehicles (AEVs), is experiencing high employee turnover. This challenge disrupts workflows, hinders productivity, and leads to significant financial burdens due to recruitment and onboarding costs. Therefore, the HR department at Salifort Motors wants to take some initiatives to improve employee satisfaction levels at the company. It is important to understand what factors contribute to their leaving in order to help the company develop appropiate solutions to increase employee retention.

Data Understanding

The employee survey data came from Kaggle. The data consisted of approximately 14,999 different employee's self-reported survey responses and 10 features. The features included information on the level of employee satisfaction and performance score, number of project, average monthly hours, time spent at the company, promotion in the last 5 years, department, and salary rate. The bar chart below shows the breakdown of how many leavers and non-leavers exist in the data set.

Bar Plot Showing Leavers vs Non-Leavers

Modeling and Evaluation

A random forest model comprising 100 decision trees was used to determine feature importance in who would leave the company or not. The below plot shows that the number of projects, average monthly hours, the last evaluation score were the top 3 most important factors in determining a leaver from a non-leaver. The overall model performed with 91% recall and 92% precision.ย 

Bar Plot Showing Feature Importances from Random Forest

Conclusion

This model can help Salifort Motors predict employees will leave the company or not by analyzing various factors through a parametric model. In the future work, to improve the model's effectiveness and minimize data leakage, removing the last evaluation score can be considered. Additionally, incorporating time-varying features related to employee behavior could further enhance the model's ability to address Salifort Motors' business problem.

salifort-employee-churn-prediction's People

Contributors

thanakorntha avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.