search

chengkai-liu / Mamba4Rec

[RelKD'24] Mamba4Rec: Towards Efficient Sequential Recommendation with Selective State Space Models
https://arxiv.org/abs/2403.03900
MIT License
85 stars 3 forks source link

Code #7

Closed hzau96yhz closed 7 months ago

hzau96yhz commented 7 months ago

Hi! Good work! What does the following line of code do?It seems that self.gather_indexes are not defined https://github.com/chengkai-liu/Mamba4Rec/blob/d4d1fd64389f55684dfd62515ced4663f92a7476/mamba4rec.py#L65

chengkai-liu commented 7 months ago

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