Skip to content

Instantly share code, notes, and snippets.

@NMZivkovic

NMZivkovic/split-data.py

Created April 21, 2019 18:30
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 NMZivkovic/779388c7005fbbac74ef67e1d195ae9b to your computer and use it in GitHub Desktop.
Save NMZivkovic/779388c7005fbbac74ef67e1d195ae9b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
split-data.py
output_data = data["Species"]
input_data = data.drop("Species",axis=1)
X_train, X_test, y_train, y_test = train_test_split(input_data, output_data, test_size=0.3, random_state=42)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment