(self, x, edge_index, edge_weigt=None, batch=None)
| 266 | |
| 267 | |
| 268 | def get_loss(self, x, edge_index, edge_weigt=None, batch=None): |
| 269 | |
| 270 | mask_x, recon_x = self.forward(x, edge_index, edge_weigt, batch) |
| 271 | loss = sce_loss(mask_x, recon_x, alpha=2) |
| 272 | |
| 273 | return loss |
| 274 | |
| 275 | def embed(self, x, edge_index, edge_weigt=None, batch=None): |
| 276 |