Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Predict Employee Turnover #289

Merged
merged 1 commit into from
Oct 12, 2024
Merged

Conversation

ramana2074
Copy link
Contributor

Related Issues or bug

-There was an issue related to class imbalance in the employee churn dataset, which could lead to biased model predictions, especially towards the majority class. This was addressed through stratified sampling during train-test splits and additional analysis of model performance using metrics like recall and F1-score.
Fixes: #246

Proposed Changes

  • Info about Changes:
    Added data preprocessing steps to handle categorical variables by encoding them into numerical features.
    Implemented class balance visualization using Yellowbrick to detect class imbalance and influence our sampling strategy.
    Built and trained a Decision Tree Classifier and Random Forest Classifier with interactive controls for hyperparameter tuning using the interact function.
    Evaluated models using training and validation accuracy, alongside feature importance plots to identify key factors influencing employee churn.

Additional Info

  • This fix also addresses the challenge of overfitting in decision trees by introducing Random Forests, which reduce variance and improve model generalization. The class imbalance issue was mitigated through careful evaluation of recall and F1-score in addition to accuracy.

Copy link

👋 Thank you for opening this pull request! We appreciate your contribution to improving this project. Your PR is under review, and we'll get back to you shortly.
Don't forget to mention the issue you solved!.

To help move the process along, please tag @UppuluriKalyani, @Neilblaze, and @SaiNivedh26 for a faster review!

@UppuluriKalyani UppuluriKalyani merged commit aa39dbd into UppuluriKalyani:main Oct 12, 2024
3 checks passed
Copy link

🎉🎉 Thank you for your contribution! Your PR #289 has been merged! 🎉🎉

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Predict-Employee-Turnover-with-scikit-learn
2 participants