Avdhesh-Varshney commented 5 months ago

Pull Request for ML-Crate 💡

Issue Title: Nurse Stress Prediction

Info about the related issue (Aim of the project) : The aim of this project is to predict the stress on nurse based on the given dataset.
Name: Avdhesh Varshney
GitHub ID: https://github.com/Avdhesh-Varshney
Idenitfy yourself: IWOC'24 Contributor

Closes: #517

Load the dataset which contains 11509051 entries in it and having 9 columns in it.
Checked for missing values and cleaned the data accordingly.
Analyzed the data, found insights and visualized them accordingly.
Plotting heatmap using correlation and checking the relation between different features.
Performing PCA to reduce the number of features and normalize the data to remove the effect of outliers.
Found detailed insights of different columns with target variable using plotting libraries.
Train the datasets by different models and saves their accuracies into a dataframe.
Visualize the accuracy and finalize the best fitted model.

[x] My code follows the guidelines of this project.
[x] I have performed a self-review of my own code.
[x] I have commented my code, particularly wherever it was hard to understand.
[x] I have made corresponding changes to the documentation.
[x] My changes generate no new warnings.
[x] I have added things that prove my fix is effective or that my feature works.
[x] Any dependent changes have been merged and published in downstream modules.

github-actions[bot] commented 5 months ago

Our team will soon review your PR. Thanks @Avdhesh-Varshney :)

Avdhesh-Varshney commented 5 months ago

@abhisheks008 please review this pr and requested to merge this pr under IWOC'24