Python PyTorch movielens_20m用法及代码示例

本文简要介绍python语言中 torchrec.datasets.movielens.movielens_20m 的用法。

用法:

torchrec.datasets.movielens.movielens_20m(root: str, *, include_movies_data: bool = False, row_mapper: typing.Optional[typing.Callable[[typing.List[str]], typing.Any]] = <function _default_row_mapper>, **open_kw) → torch.utils.data.dataset.IterDataPipe

MovieLens 20M 数据集：param root：包含MovieLens 20M 数据集文件的根目录的本地路径。 :type root: str :param include_movies_data: 如果为 True，则将电影数据添加到每一行。 :type include_movies_data: bool :param row_mapper: 应用于每个分割线的函数。 :type row_mapper: 可选[Callable[[List[str]], Any]] :param open_kw: 传递给 iopath.common.file_io.PathManager.open 底层调用的选项。

例子：

datapipe = movielens_20m("/home/datasets/ml-20")
datapipe = dp.iter.Batch(datapipe, 100)
datapipe = dp.iter.Collate(datapipe)
batch = next(iter(datapipe))