Skip to content

Instantly share code, notes, and snippets.

@crawles

crawles/load_titanic.py

Last active March 5, 2019 02:28
Show Gist options
  • Save crawles/9b0e472523f5cd3da270e1c5b41273b8 to your computer and use it in GitHub Desktop.
Save crawles/9b0e472523f5cd3da270e1c5b41273b8 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
load_titanic.py
import numpy as np
import pandas as pd
import tensorflow as tf
tf.enable_eager_execution()
# Load dataset.
dftrain = pd.read_csv('https://storage.googleapis.com/tf-datasets/titanic/train.csv')
dfeval = pd.read_csv('https://storage.googleapis.com/tf-datasets/titanic/eval.csv')
y_train = dftrain.pop('survived')
y_eval = dfeval.pop('survived')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment