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

Create gbm.py- Solving issue #128 #154

Merged
merged 1 commit into from
Jun 3, 2024

Conversation

Aditi22Bansal
Copy link
Contributor

Solved issue #128
Data Loading and Basic Information: Loading the dataset and displaying basic information.
Missing Values Check: Checking for missing values in the dataset.
Class Distribution: Checking and visualizing the class distribution to identify any imbalance.
Summary Statistics: Displaying summary statistics of the dataset.
Label Encoding: Encoding categorical features using LabelEncoder.
Feature and Target Definition: Separating features and target variable.
Preprocessing: Defining preprocessing steps for numerical and categorical features.
Pipeline and Model Definition: Creating a pipeline that includes preprocessing and the Gradient Boosting model.
Train-Test Split: Splitting the data into training and testing sets.
Hyperparameter Tuning: Using GridSearchCV to find the best hyperparameters for the model.
Model Evaluation: Evaluating the model on the test set and displaying metrics.
ROC Curve: Plotting the ROC curve to evaluate the model's performance.
Feature Importance: Displaying and plotting the feature importances.
Model Saving: Saving the best model using joblib.

Attaching screenshots of the results
image
image
image
image
image
image
image

Copy link

github-actions bot commented Jun 3, 2024

Thank you for submitting your pull request! 🙌 We'll review it as soon as possible. In the meantime, please ensure that your changes align with our CONTRIBUTING.md. If there are any specific instructions or feedback regarding your PR, we'll provide them here. Thanks again for your contribution! 😊

@Aditi22Bansal
Copy link
Contributor Author

@sanjay-kv
Kindly review this PR and approve
Thanks

@sanjay-kv sanjay-kv merged commit 5fca2ac into recodehive:main Jun 3, 2024
2 checks passed
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.

2 participants