Skip to content

Instantly share code, notes, and snippets.

@bharatc9530

bharatc9530/split.py

Created August 4, 2020 08:01
Show Gist options
  • Save bharatc9530/bfc6b41c6a83921d0dbd3366130dd9f6 to your computer and use it in GitHub Desktop.
Save bharatc9530/bfc6b41c6a83921d0dbd3366130dd9f6 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
split.py
docs = df['review']
labels = array(df['sentiment'])
from sklearn.model_selection import train_test_split
X_train, X_test , y_train, y_test = train_test_split(docs, labels , test_size = 0.40)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment