Skip to content

This was a 3-month long program where we learned about various data science techniques and made a capstone project in the end.

Notifications You must be signed in to change notification settings

thepankj/Employee-Attrition-Prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Employee-Attrition-Prediction

This was the Capstone Project of a 3 month long program at IIT Guwahati where we learned about various data science techniques. The IPYNB file shows the various EDA I did as well as data preprocessing for the prediction.


Features

The various features in the dataset are

Features


There are no missing values

Missing values heatmap


And the dataset is balanced

Balanced or Imbalanced


Plotted the distributions of the some the features

Distributions


Before applying the models, I checked the importances of the features using Extra Tree Classifier

Feature Importances


Used RandomizedSearchCV on Logistic Regression which gave an accuracy of 79.37% and SVC gave an accuracy of 80.3%

About

This was a 3-month long program where we learned about various data science techniques and made a capstone project in the end.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published