Skip to content

kennethcandersen/Machine-Learning-to-Predict-Employee-Retention

Repository files navigation

The Talent Retention Predictor App

Machine Learning Model to predict the retention of employees

Team Members

Project Goal

Create a Machine Learning Model to predict the retention of employees using a data set from IBM.

How to run the app

Click on the following link to check the app,Data Analysis in Tableau and the Machine Learning and Deep Learnig models :

  1. Fully deployed online version to go directly to the app online.

  2. Run locally:

    1. Clone this repository
    2. In the main repository folder, run "python3 app.py" in your terminal
    3. Open http://127.0.0.1:5000/ in a browser and the website should launch.
    4. Browse the website...

Original data Source 📁

The original data can be found here at Kaggle

ETL Process

Extract

  • Download the original HR dataset from IBM

Transform

  • Run numerous Machine Learning trials using sklearn, TensorFlow and Keras in Jupyter Notebook
  • Create executive summary data visualizations in Tableau

Load

  • Use Flask to run the application online.
  • Create website using Bootstrap 5.
  • Create interactive retention predictor dashboard using Javascript, which calls a route in Flask to produce the machine learning algorithm in Python.
  • Embed our Tableau visualizations and Jupyter Notebook on informational pages within the website.

Languages & Tools Used 🖥️

  • Python
  • Jupyter Notebook
  • Sklearn, TensorFlow & Keras libraries
  • Flask
  • Javascript
  • HTML
  • CSS

About

Web-based app that uses a machine learning model to predict employee retention.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •