Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
python imbalanced data machine learning classification
from sklearn.linear_model import LogisticRegression
clf = LogisticRegression(random_state=888)
clf.fit(X_resampled, y_resampled)
y_pred = clf.predict_proba(df_test[features])[:, 1]
from sklearn.metrics import roc_auc_score
roc_auc_score(df_test['Class'], y_pred)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment