(self, pre_user, pre_item, )
| 530 | |
| 531 | |
| 532 | def forward(self, pre_user, pre_item, ): |
| 533 | # pre_user, pre_item = self.user_id_embedding.weight, self.item_id_embedding.weight |
| 534 | user_embed = self.user_trans(pre_user) |
| 535 | item_embed = self.user_trans(pre_item) |
| 536 | |
| 537 | return user_embed, item_embed |
| 538 | # return pre_user, pre_item |
| 539 | |
| 540 | def init_user_item_embed(self, pre_u_embed, pre_i_embed): |
| 541 | self.user_id_embedding = nn.Embedding.from_pretrained(pre_u_embed, freeze=False) |