Skip to content

Instantly share code, notes, and snippets.

@animesh-agarwal

animesh-agarwal/train_test_split.py

Last active October 1, 2018 17:36
Show Gist options
  • Save animesh-agarwal/0bfaaf1740fa3bb2a3be61f60f46c97a to your computer and use it in GitHub Desktop.
Save animesh-agarwal/0bfaaf1740fa3bb2a3be61f60f46c97a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
train_test_split.py
from sklearn.model_selection import train_test_split
X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size = 0.2, random_state=5)
print(X_train.shape)
print(X_test.shape)
print(Y_train.shape)
print(Y_test.shape)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment