Here I used Decision tree and Random forest to see whether an employee is going to leave the company or not. Finally I added Yellowbrick, a powerful library to plot some nice curve. To make it more useful I added another intersting feature, i.e. intractive plotting for both DT and RF classifier. Which can calculate the turnover with more accuracy because you can change the height, no of estimator in the fly.
I added the source code and dataset as well. Necessary instructions and little theory are provided. The DT model was a little overfitted so finally I used RF which did a good job in predicting this. Finally I Plotted a ROC curve to see how well our model is doing.
-
Notifications
You must be signed in to change notification settings - Fork 0
prabormukherjee/Employee_turnover
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Using Random forest to predict employee turnover
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published