Skip to content

Instantly share code, notes, and snippets.

@betterdatascience

betterdatascience/roc_auc.py

Created December 8, 2020 08:06
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save betterdatascience/3ac57d403524b795a4cfa5b7919a8c8c to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save betterdatascience/3ac57d403524b795a4cfa5b7919a8c8c to your computer and use it in GitHub Desktop.
Download ZIP
004_roc_auc
Raw
roc_auc.py
from sklearn.model_selection import train_test_split
X = df.drop('quality', axis=1)
y = df['quality']
X_train, X_test, y_train, y_test = train_test_split(
X, y, test_size=0.25, random_state=42
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment