Skip to content

Commit 42091f1

Browse files
Update hstu.py
Fix broken link
1 parent 847628b commit 42091f1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

recml/layers/keras/hstu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def default_bucketization(tensor: Tensor, bucket_size: float = 0.301):
3535
"""Default bucketization function for RelativeBucketedTimeAndPositionBasedBias.
3636
3737
From the original paper's code base:
38-
https://github.com/facebookresearch/generative-recommenders/blob/main/modeling/sequential/hstu.py#L79
38+
https://github.com/meta-recsys/generative-recommenders/blob/main/generative_recommenders/research/modeling/sequential/hstu.py#L87
3939
4040
Args:
4141
tensor: The tensor of timestamps to bucketize.

0 commit comments

Comments
 (0)