(self, user_x, item_x, ui_graph, iu_graph)
| 385 | |
| 386 | |
| 387 | def forward(self, user_x, item_x, ui_graph, iu_graph): |
| 388 | # # x, support = inputs |
| 389 | # # user_x, item_x = self.layers((user_x, item_x, ui_graph, iu_graph)) |
| 390 | # for i in range(args.student_n_layers): |
| 391 | # user_x, item_x = self.layer_list[i](user_x, item_x, ui_graph, iu_graph) |
| 392 | # return user_x, item_x |
| 393 | |
| 394 | return self.trans_user(user_x), self.trans_item(item_x) |
| 395 | # self.user_id_embedding = nn.Embedding.from_pretrained(user_x, freeze=True) |
| 396 | # self.item_id_embedding = nn.Embedding.from_pretrained(item_x, freeze=True) |
| 397 | # return self.user_id_embedding.weight, self.item_id_embedding.weight |
| 398 | |
| 399 | def l2_loss(self): |
| 400 | layer = self.layers.children() |
nothing calls this directly
no outgoing calls
no test coverage detected