Skip to content

Instantly share code, notes, and snippets.

@FranciscusRenatus

FranciscusRenatus/splitting_the_dataset.py

Created June 20, 2017 17:48
Show Gist options
  • Save FranciscusRenatus/5ef1d2ef0b04ebd679b51c4ad574987f to your computer and use it in GitHub Desktop.
Save FranciscusRenatus/5ef1d2ef0b04ebd679b51c4ad574987f to your computer and use it in GitHub Desktop.
Download ZIP
Raw
splitting_the_dataset.py
# Splitting the dataset into train and test set
from sklearn.cross_validation import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.2, random_state = 0)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment