Merge pull request #31 from aimclub/fairness

Fairness
aimclub · Jan 17, 2024 · 28cb607 · 28cb607
2 parents 7af183b + 7c71939
commit 28cb607
Show file tree

Hide file tree

Showing 7 changed files with 23,481 additions and 1 deletion.
diff --git a/stable_gnn/embedding/sampling/abstract_samplers.py b/stable_gnn/embedding/sampling/abstract_samplers.py
@@ -96,7 +96,7 @@ def _sample_negative(self, batch: Tensor, num_negative_samples: int) -> Tensor:
         Sample negative edges for batch of nodes
 
         :param batch: (Batch): Nodes for negative sampling
-        :param num_negative_samples: (int): number of negative samples for each edge
+        :param num_negative_samples: (int): Number of negative samples for each edge
         :return: (Tensor): Negative samples
         """
         a, _ = subgraph(batch, self.data.edge_index.to(self.device))