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.
2 parents 847628b + 42091f1 commit 61c08caCopy full SHA for 61c08ca
recml/layers/keras/hstu.py
@@ -35,7 +35,7 @@ def default_bucketization(tensor: Tensor, bucket_size: float = 0.301):
35
"""Default bucketization function for RelativeBucketedTimeAndPositionBasedBias.
36
37
From the original paper's code base:
38
- https://github.com/facebookresearch/generative-recommenders/blob/main/modeling/sequential/hstu.py#L79
+ https://github.com/meta-recsys/generative-recommenders/blob/main/generative_recommenders/research/modeling/sequential/hstu.py#L87
39
40
Args:
41
tensor: The tensor of timestamps to bucketize.
0 commit comments