Skip to content

Instantly share code, notes, and snippets.

@aniruddha27

aniruddha27/AUC-ROC1.py

Last active Apr 11, 2020
Embed
What would you like to do?
from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split
# generate two class dataset
X, y = make_classification(n_samples=1000, n_classes=2, n_features=20, random_state=27)
# split into train-test sets
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=27)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment