Breaking News: Grepper is joining You.com. Read the official announcement!

sample hyperparameter tuning with grid search cv

josh.ipynb answered on June 11, 2022 Popularity 9/10 Helpfulness 2/10

answer sample hyperparameter tuning with grid search cv

sample hyperparameter tuning with grid search cv

Comment

# Define params_dt
params_dt = {
             'max_depth': [2, 3, 4],
             'min_samples_leaf': [0.12, 0.14, 0.16, 0.18]
            }
# Import GridSearchCV
from sklearn.model_selection import GridSearchCV

# Instantiate grid_dt
grid_dt = GridSearchCV(estimator=...base_model...,
                       param_grid=params_dt,
                       scoring='roc_auc',
                       cv=5,
                       n_jobs=-1)

#Train the model
grid_dt.fit(X_train, y_train)

# Import roc_auc_score from sklearn.metrics 
from sklearn.metrics import roc_auc_score

# Extract the best estimator
best_model = grid_dt.best_estimator_

# Predict the test set probabilities of the positive class
y_pred_proba = best_model.predict_proba(X_test)[:,1]

# Compute test_roc_auc
test_roc_auc = roc_auc_score(y_test, y_pred_proba)

# Print test_roc_auc
print('Test set ROC AUC score: {:.3f}'.format(test_roc_auc))

Popularity 9/10 Helpfulness 2/10 Language python

Source: Grepper

Tags: grid-search python

Link to this answer
Share Copy Link

Contributed on Jun 11 2022

josh.ipynb

0 Answers Avg Quality 2/10

sample hyperparameter tuning with grid search cv

Contents

More Related Answers

sample hyperparameter tuning with grid search cv

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.