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:
DatasetThe 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