Skip to content

Instantly share code, notes, and snippets.

@lisanka93
Created August 3, 2020 16:25
Show Gist options
  • Save lisanka93/1edf3331db0f184393f3b17e71625e6b to your computer and use it in GitHub Desktop.
Save lisanka93/1edf3331db0f184393f3b17e71625e6b to your computer and use it in GitHub Desktop.
splitting data into train and test set
X_train, X_test, y_train, y_test = train_test_split(
covid_data['prep_arg'],
covid_data['concern'],
test_size=0.2,
random_state=50
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment