This project requires Python 3.6 and the following Python libraries installed:
NumPy
Pandas
matplotlib
scikit-learn
You will also need to have software installed to run and execute an iPython Notebook
We recommend to install Anaconda, a pre-packaged Python distribution that contains all of the necessary libraries and software for this project.
Code is provided in the CapstoneProject.ipynb notebook file. It uses the HR_comma_sep.csv dataset file. Code uses numpy, python, matplotlib and scikit learn libraries. Code uses classification technique of Supervised Learning to train the model. Code has implemented ADABoost Classifier, Gradient Descent Classifier, SVM and Stochastic gradient Descent to calculate performance based on the different methods of training using different techniques.
In a terminal or command window, navigate to the top-level project directory capstone_project/ (that contains this README) and run one of the following commands:
ipython notebook CapstoneProject.ipynb
or
jupyter notebook CapstoneProject.ipynb
This will open the iPython Notebook software and project file in your browser.
This data set contains total of 14999 rows and 10 columns. Target variable (left) is imbalance dataset. It contains 3751 records of employee who have left the company. 11428 records of employee who stayed in the company. Here we are trying to predict employee who can quit. Dataset has been extracted from Kaggle.
Dataset Link: https://www.kaggle.com/ludobenistant/hr-analytics
- satisfaction_level : Level of Satisfaction
- last_evaluation :Time since Last performance Evaluation
- number_project :Number of Project completed while at work
- average_montly_hours : Average monthly hours at workplace
- time_spend_company : Number of years spent in the company
- Work_accident : Whether the employee had a workplace accident
- promotion_last_5years: Wehter employee was promoted in last 5 years
- sales : Department they work for
- Salary : Relative level of Salary(high)
- left : Whether employee left the workplace or not (0 - stayed and 1 - Left)