torch_frame.datasets.Titanic

class Titanic(root: str)[来源]

基础类:Dataset

来自泰坦尼克号 Kaggle 竞赛的泰坦尼克号数据集。 泰坦尼克号数据集被称为表格学习的 MNIST 等价物。 目标是使用乘客数据(性别、年龄等)预测哪些乘客幸存下来。

统计:

#行数

#cols(数值)

#cols(分类)

#classes

任务

缺失值比例

891

4

3

2

二元分类

8.8%