Skip to content

Instantly share code, notes, and snippets.

@mohdsanadzakirizvi

mohdsanadzakirizvi/bert_split.py

Created September 23, 2019 17:59
Show Gist options
  • Save mohdsanadzakirizvi/f1c85bdba3f2b0e11b7395e8fa3bddbe to your computer and use it in GitHub Desktop.
Save mohdsanadzakirizvi/f1c85bdba3f2b0e11b7395e8fa3bddbe to your computer and use it in GitHub Desktop.
Download ZIP
Raw
bert_split.py
from sklearn.model_selection import train_test_split
# split into training and validation sets
X_tr, X_val, y_tr, y_val = train_test_split(train.clean_text, train.label, test_size=0.25, random_state=42)
print('X_tr shape:',X_tr.shape)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment