Skip to content

Instantly share code, notes, and snippets.

@liannewriting

liannewriting/auc_oversampling_smote.py

Created October 13, 2021 17:19
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save liannewriting/947e8f3d22aa54e927a8e1f41e69e5b3 to your computer and use it in GitHub Desktop.
Save liannewriting/947e8f3d22aa54e927a8e1f41e69e5b3 to your computer and use it in GitHub Desktop.
Download ZIP
python imbalanced data machine learning classification
Raw
auc_oversampling_smote.py
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