Skip to content

Instantly share code, notes, and snippets.

@liannewriting

liannewriting/train_test_split.py

Last active July 30, 2020 20:13
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 liannewriting/4abd610e1f8a0ef95da324d8ecda1162 to your computer and use it in GitHub Desktop.
Save liannewriting/4abd610e1f8a0ef95da324d8ecda1162 to your computer and use it in GitHub Desktop.
Download ZIP
random forest machine learning
Raw
train_test_split.py
breast_cancer = load_breast_cancer()
X_train, X_test, y_train, y_test = train_test_split(breast_cancer.data, breast_cancer.target, test_size=0.2, stratify=breast_cancer.target, random_state=random_seed)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment