Drop self-loops removal and addition

pyg-team · Jun 29, 2022 · c2c77f1 · c2c77f1
1 parent 63cd463
commit c2c77f1
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/torch_geometric/nn/models/autoencoder.py b/torch_geometric/nn/models/autoencoder.py
@@ -95,9 +95,6 @@ def recon_loss(self, z, pos_edge_index, neg_edge_index=None):
             self.decoder(z, pos_edge_index, sigmoid=True) + EPS).mean()
 
         if neg_edge_index is None:
-            # Do not include self-loops in negative samples
-            pos_edge_index, _ = remove_self_loops(pos_edge_index)
-            pos_edge_index, _ = add_self_loops(pos_edge_index)
             neg_edge_index = negative_sampling(pos_edge_index, z.size(0))
         neg_loss = -torch.log(1 -
                               self.decoder(z, neg_edge_index, sigmoid=True) +