Skip to content

Instantly share code, notes, and snippets.

@liannewriting

liannewriting/auc_undersampling_k_means.py

Created October 13, 2021 17:22
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/1873ac2ab5a0237ac9a7b5fc69e9ffa4 to your computer and use it in GitHub Desktop.
Save liannewriting/1873ac2ab5a0237ac9a7b5fc69e9ffa4 to your computer and use it in GitHub Desktop.
Download ZIP
python imbalanced data machine learning classification
Raw
auc_undersampling_k_means.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