Skip to content

Instantly share code, notes, and snippets.

@aravindpai

aravindpai/split.py

Created May 27, 2019 06:34
Show Gist options
  • Save aravindpai/f3632c4a87b769848c4dadb1421bceac to your computer and use it in GitHub Desktop.
Save aravindpai/f3632c4a87b769848c4dadb1421bceac to your computer and use it in GitHub Desktop.
Download ZIP
Raw
split.py
from sklearn.model_selection import train_test_split
x_tr,x_val,y_tr,y_val=train_test_split(data['cleaned_text'],data['cleaned_summary'],test_size=0.1,random_state=0,shuffle=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment