Skip to content

Instantly share code, notes, and snippets.

@yashprakash13

yashprakash13/split.py

Created December 26, 2021 16:03
Show Gist options
  • Save yashprakash13/b836eeb980df1e8745a2b674d7707e8f to your computer and use it in GitHub Desktop.
Save yashprakash13/b836eeb980df1e8745a2b674d7707e8f to your computer and use it in GitHub Desktop.
Download ZIP
Raw
split.py
from sklearn.model_selection import train_test_split
X, y = df.drop('DEATH_EVENT', axis=1), df['DEATH_EVENT']
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3)
X_train.shape, X_test.shape, y_train.shape, y_test.shape
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment