Skip to content

Instantly share code, notes, and snippets.

@AyishaR

AyishaR/spam_ham_train_val_test.py

Created January 22, 2021 16:25
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save AyishaR/55c42ad239e04a3cd7a3687b53d50884 to your computer and use it in GitHub Desktop.
Save AyishaR/55c42ad239e04a3cd7a3687b53d50884 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
spam_ham_train_val_test.py
# Splitting into train, val and test set -- 80-10-10 split
# First, an 80-20 split
train_df, val_test_df = train_test_split(df, test_size = 0.2, random_state = 113)
# Then split the 20% into half
val_df, test_df = train_test_split(val_test_df, test_size = 0.5, random_state = 113)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment