People analytics - Employee attrition prediction

Build following predicative models using the fictional IBM data set which contains 1470 employee attrition records.

• Decision Tree
• Random Forest
• XGBoost

Install

This project requires Python 3.0+ and the following Python libraries installed:

• NumPy
• Pandas
• matplotlib
• seaborn
• scikit-learn
• XGBoost

You will also need to have software installed to run and execute an iPython Notebook

Code

Code is provided in the notebook ibm_hr_analytics.ipynb notebook file.

Run

In a terminal or command window, navigate to the top-level project directory People-analytics-employee-attrition-prediction/ and run one of the following commands:

ipython notebook ibm_hr_analytics.ipynb jupyter notebook ibm_hr_analytics.ipynb

This will open the iPython Notebook software and project file in your browser.

Data

fictional IBM data set which contains 1470 employee attrition records.