Skip to content

Instantly share code, notes, and snippets.

@FeryET

FeryET/dataset_split.py

Created August 26, 2020 11:00
Show Gist options
  • Save FeryET/1621a1ca4aff41f01adb108923e89248 to your computer and use it in GitHub Desktop.
Save FeryET/1621a1ca4aff41f01adb108923e89248 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
dataset_split.py
docs_train, docs_test, y_train, y_test = train_test_split(docs, y, test_size=0.1, shuffle=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment