Skip to content

Instantly share code, notes, and snippets.

@kusal1990
Created June 2, 2022 18:24
Show Gist options
  • Save kusal1990/53005ee409bf8a2760631e923eb3fdff to your computer and use it in GitHub Desktop.
Save kusal1990/53005ee409bf8a2760631e923eb3fdff to your computer and use it in GitHub Desktop.
# Parameters to tune for LR model
params = {'C': [10**x for x in range(-5,6)]}
# Create a custom (MCC) metric for evaluation of the model performance while
# hyperparameter tuning the XGBoost model
mcc = make_scorer(matthews_corrcoef, greater_is_better=True)
# Create an XGBoost classifier object with log-loss as the loss function to minimize
log_clf = LogisticRegression(random_state=42, class_weight='balanced')
# Perform stratified 5-fold cross validation
grid_clf = GridSearchCV(log_clf, params, scoring=mcc, cv=5, return_train_score=True)
# Fit the model
grid_clf .fit(X_train, y_train)
print(f"Best CV score: {grid_clf.best_score_}")
@kusal1990
Copy link
Author

ok

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