torch_frame.datasets.Titanic
- class Titanic(root: str)[source]
Bases:
DatasetThe 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%