(self, x, edge_index, edge_weigt=None, batch=None)
| 223 | |
| 224 | |
| 225 | def forward(self, x, edge_index, edge_weigt=None, batch=None): |
| 226 | # Mask |
| 227 | mask_x, mask_nodes = self.encding_mask(x) |
| 228 | h = self.encoder(mask_x, edge_index, edge_weigt, batch) |
| 229 | h = self.encoder_to_decoder(h) |
| 230 | |
| 231 | # Re-mask |
| 232 | h[mask_nodes] = 0 |
| 233 | recon = self.decoder(h, edge_index, edge_weigt, batch) |
| 234 | |
| 235 | return x[mask_nodes], recon[mask_nodes] |
| 236 | |
| 237 | |
| 238 | def encding_mask(self, x): |