torch_frame.datasets.Movielens1M

class Movielens1M(root: str, col_to_text_embedder_cfg: Optional[Union[dict[str, torch_frame.config.text_embedder.TextEmbedderConfig], TextEmbedderConfig]] = None)[source]

Bases: Dataset

The MovieLens 1M rating dataset, assembled by GroupLens Research from the MovieLens web site, consisting of movies (3,883 nodes) and users (6,040 nodes) with approximately 1 million ratings between them.

STATS:

#Users

#Items

#User Field

#Item Field

#Samples

6040

3952

5

3

1000209