We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
torchrec
1 parent 649c937 commit 778b47cCopy full SHA for 778b47c
examples/bert4rec/bert4rec_main.py
@@ -445,6 +445,7 @@ def main(argv: List[str]) -> None:
445
446
world_size = dist.get_world_size()
447
448
+ # pyre-fixme[16]: Module `data` has no attribute `bert4rec_movielens_datasets`.
449
raw_data = get_raw_dataframe(
450
args.dataset_name,
451
args.random_user_count,
@@ -454,6 +455,7 @@ def main(argv: List[str]) -> None:
454
455
args.dataset_path,
456
)
457
458
459
df = Bert4RecPreprocsser(
460
raw_data,
461
args.min_rating,
0 commit comments