当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch movielens_25m用法及代码示例


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

用法:

torchrec.datasets.movielens.movielens_25m(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 25M 数据集:param root:包含MovieLens 25M 数据集文件的根目录的本地路径。 :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_25m("/home/datasets/ml-25")
datapipe = dp.iter.Batch(datapipe, 100)
datapipe = dp.iter.Collate(datapipe)
batch = next(iter(datapipe))

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torchrec.datasets.movielens.movielens_25m。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。