Skip to content

Instantly share code, notes, and snippets.

@liannewriting

liannewriting/auc_undersampling_pandas.py

Created October 13, 2021 17:20
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/3e1ff7d5d31327150444897c4839849a to your computer and use it in GitHub Desktop.
Save liannewriting/3e1ff7d5d31327150444897c4839849a to your computer and use it in GitHub Desktop.
Download ZIP
python imbalanced data machine learning classification
Raw
auc_undersampling_pandas.py
from sklearn.linear_model import LogisticRegression
clf = LogisticRegression(random_state=888)
clf.fit(df_train_undersample[features], df_train_undersample['Class'])
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