torch_frame.datasets.AmazonFineFoodReviews

class AmazonFineFoodReviews(root: str, text_stype: stype = stype.text_embedded, col_to_text_embedder_cfg: Optional[Union[dict[str, torch_frame.config.text_embedder.TextEmbedderConfig], TextEmbedderConfig]] = None, col_to_text_tokenizer_cfg: Optional[Union[dict[str, torch_frame.config.text_tokenizer.TextTokenizerConfig], TextTokenizerConfig]] = None)[source]

Bases: Dataset

The Amazon Fine Food Reviews dataset. It consists of reviews of fine foods from amazon.

Parameters:

text_stype (torch_frame.stype) – Text stype to use for text columns in the dataset. (default: torch_frame.text_embedded)

STATS:

#rows

#cols (numerical)

#cols (categorical)

#cols (text)

#classes

Task

Missing value ratio

568,454

2

3

2

5

multiclass_classification

0.0%