torch_frame.datasets.DiamondImages

class DiamondImages(root: str, col_to_image_embedder_cfg: torch_frame.config.image_embedder.ImageEmbedderConfig | dict[str, torch_frame.config.image_embedder.ImageEmbedderConfig])

Bases: Dataset

The Diamond Images dataset from Kaggle. The target is to predict colour of each diamond.

STATS:

#rows	#cols (numerical)	#cols (categorical)	#cols (image)	#classes	Task	Missing value ratio
48,764	4	7	1	23	multiclass_classification	0.167%