mohdsanadzakirizvi/bert_split.py

## 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)
	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)