Skip to content

Instantly share code, notes, and snippets.

@tommydangerous

tommydangerous/download_and_split_data.py

Created June 11, 2021 05:22
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save tommydangerous/07b049dddee5bba1ad1c6e246e7a05a2 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save tommydangerous/07b049dddee5bba1ad1c6e246e7a05a2 to your computer and use it in GitHub Desktop.
Download ZIP
download_and_split_data
Raw
download_and_split_data.py
from sklearn.model_selection import train_test_split
import pandas as pd
df = pd.read_csv('/content/titanic_survival.csv')
label_feature_name = 'Survived'
X = df.drop(columns=[label_feature_name])
y = df[label_feature_name]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment