Tune in to the first Women in ML Symposium this Tuesday, October 19 at 9am PST Register now

titanic

  • Description:

Dataset describing the survival status of individual passengers on the Titanic. Missing values in the original dataset are represented using ?. Float and int missing values are replaced with -1, string missing values are replaced with 'Unknown'.

Split Examples
'train' 1,309
  • Features:
FeaturesDict({
    'age': tf.float32,
    'boat': tf.string,
    'body': tf.int32,
    'cabin': tf.string,
    'embarked': ClassLabel(shape=(), dtype=tf.int64, num_classes=4),
    'fare': tf.float32,
    'home.dest': tf.string,
    'name': tf.string,
    'parch': tf.int32,
    'pclass': ClassLabel(shape=(), dtype=tf.int64, num_classes=3),
    'sex': ClassLabel(shape=(), dtype=tf.int64, num_classes=2),
    'sibsp': tf.int32,
    'survived': ClassLabel(shape=(), dtype=tf.int64, num_classes=2),
    'ticket': tf.string,
})
  • Supervised keys (See as_supervised doc): ({'name': 'name', 'sex': 'sex', 'home.dest': 'home.dest', 'fare': 'fare', 'pclass': 'pclass', 'age': 'age', 'body': 'body', 'embarked': 'embarked', 'sibsp': 'sibsp', 'ticket': 'ticket', 'boat': 'boat', 'cabin': 'cabin', 'parch': 'parch'}, 'survived')

  • Figure (tfds.show_examples): Not supported.

  • Examples (tfds.as_dataframe):

  • Citation:
@ONLINE {titanic,
author = "Frank E. Harrell Jr., Thomas Cason",
title  = "Titanic dataset",
month  = "oct",
year   = "2017",
url    = "https://www.openml.org/d/40945"
}