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))