torch_frame.datasets.DiamondImages

class DiamondImages(root: str, col_to_image_embedder_cfg: ImageEmbedderConfig | dict[str, ImageEmbedderConfig])

基础类：Dataset

来自Kaggle的钻石图像数据集。目标是预测每颗钻石的颜色。

统计:

#行数	#cols（数值）	#cols（分类）	#cols (图像)	#classes	任务	缺失值比例
48,764	4	7	1	23	多类分类	0.167%