Skip to content

Instantly share code, notes, and snippets.

@codeperfectplus

codeperfectplus/split_the_data.py

Last active June 28, 2020 03:11
Show Gist options
  • Save codeperfectplus/f2173c6469d608c609f55091576d299c to your computer and use it in GitHub Desktop.
Save codeperfectplus/f2173c6469d608c609f55091576d299c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
split_the_data.py
## Splitting the dataset into the Training set and Test set
from sklearn.model_selection import train_test_split
train_X, test_X, train_y, test_y = train_test_split(X, y, test_size = 0.3, random_state=12)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment