Skip to content

Instantly share code, notes, and snippets.

@betterdatascience

betterdatascience/roc_auc.py

Created December 8, 2020 08:07
Show Gist options
  • Save betterdatascience/85005e4d4705ad13736074af4ba7901d to your computer and use it in GitHub Desktop.
Save betterdatascience/85005e4d4705ad13736074af4ba7901d to your computer and use it in GitHub Desktop.
Download ZIP
005_roc_auc
Raw
roc_auc.py
from sklearn.linear_model import LogisticRegression
from sklearn.tree import DecisionTreeClassifier
from sklearn.ensemble import RandomForestClassifier
from xgboost import XGBClassifier
model_lr = LogisticRegression().fit(X_train, y_train)
probs_lr = model_lr.predict_proba(X_test)[:, 1]
model_dt = DecisionTreeClassifier().fit(X_train, y_train)
probs_dt = model_dt.predict_proba(X_test)[:, 1]
model_rf = RandomForestClassifier().fit(X_train, y_train)
probs_rf = model_rf.predict_proba(X_test)[:, 1]
model_xg = XGBClassifier().fit(X_train, y_train)
probs_xg = model_xg.predict_proba(X_test)[:, 1]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment