chengkai-liu/Mamba4Rec

Code

hzau96yhz opened this issue · 1 comments

hzau96yhz commented

Hi! Good work!
What does the following line of code do?It seems that self.gather_indexes are not defined

Mamba4Rec/mamba4rec.py

Line 65 in d4d1fd6

seq_output = self.gather_indexes(item_emb, item_seq_len - 1)

chengkai-liu commented

gather_indexes is a function is the class of SequentialRecommender.
We import SequentialRecommender from RecBole: from recbole.model.abstract_recommender import SequentialRecommender.

  • 👍1