torch_frame.datasets.Titanic

class Titanic(root: str)[source]

Bases: Dataset

The Titanic dataset from the Titanic Kaggle competition. The Titanic dataset is known as the MNIST equivalent for tabular learning. The goal is to predict which passenger survived using passenger data (i.e. gender, age, etc).

STATS:

#rows

#cols (numerical)

#cols (categorical)

#classes

Task

Missing value ratio

891

4

3

2

binary_classification

8.8%