pyg-nightly 2.7.0.dev20250326__py3-none-any.whl → 2.7.0.dev20250327__py3-none-any.whl

{pyg_nightly-2.7.0.dev20250326.dist-info → pyg_nightly-2.7.0.dev20250327.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyg-nightly
-Version: 2.7.0.dev20250326
+Version: 2.7.0.dev20250327
 Summary: Graph Neural Network Library for PyTorch
 Keywords: deep-learning,pytorch,geometric-deep-learning,graph-neural-networks,graph-convolutional-networks
 Author-email: Matthias Fey <matthias@pyg.org>

{pyg_nightly-2.7.0.dev20250326.dist-info → pyg_nightly-2.7.0.dev20250327.dist-info}/RECORD

@@ -1,4 +1,4 @@
-torch_geometric/__init__.py,sha256=ugU889TdB5MV0NhdM4c9v0t0koXW4fduNQG40bDl-VQ,1978
+torch_geometric/__init__.py,sha256=5IQEu8YEKzY-iYyw4MYVUOKt0JMAAwPxFs7wn456Ms8,1978
 torch_geometric/_compile.py,sha256=f-WQeH4VLi5Hn9lrgztFUCSrN_FImjhQa6BxFzcYC38,1338
 torch_geometric/_onnx.py,sha256=V9ffrIKSqhDw6xUZ12lkuSfNs48cQp2EeJ6Z19GfnVw,349
 torch_geometric/backend.py,sha256=lVaf7aLoVaB3M-UcByUJ1G4T4FOK6LXAg0CF4W3E8jo,1575
@@ -589,7 +589,7 @@ torch_geometric/utils/_grid.py,sha256=1coutST2TMV9TSQcmpXze0GIK9odzZ9wBtbKs6u26D
 torch_geometric/utils/_homophily.py,sha256=1nXxGUATFPB3icEGpvEWUiuYbjU9gDGtlWpuLbtWhJk,5090
 torch_geometric/utils/_index_sort.py,sha256=FTJacmOsqgsyof7MJFHlVVdXhHOjR0j7siTb0UZ-YT0,1283
 torch_geometric/utils/_lexsort.py,sha256=chMEJJRXqfE6-K4vrVszdr3c338EhMZyi0Q9IEJD3p0,1403
-torch_geometric/utils/_negative_sampling.py,sha256=u-7oDg8luSFto-iUqsq7eC9uek6yqc25yHAZr6X_N8A,14673
+torch_geometric/utils/_negative_sampling.py,sha256=G4O572zAQgQQlVMz6ihhE13HFKEekLLYVXcYp4ZSdcQ,15521
 torch_geometric/utils/_normalize_edge_index.py,sha256=H6DY-Dzi1Psr3igG_nb0U3ZPNZz-BBDntO2iuA8FtzA,1682
 torch_geometric/utils/_normalized_cut.py,sha256=uwVJkl-Q0tpY-w0nvcHajcQYcqFh1oDOf55XELdjJBU,1167
 torch_geometric/utils/_one_hot.py,sha256=vXC7l7zudYRZIwWv6mT-Biuk2zKELyqteJXLynPocPM,1404
@@ -633,7 +633,7 @@ torch_geometric/utils/undirected.py,sha256=H_nfpI0_WluOG6VfjPyldvcjL4w5USAKWu2x5
 torch_geometric/visualization/__init__.py,sha256=PyR_4K5SafsJrBr6qWrkjKr6GBL1b7FtZybyXCDEVwY,154
 torch_geometric/visualization/graph.py,sha256=ZuLPL92yGRi7lxlqsUPwL_EVVXF7P2kMcveTtW79vpA,4784
 torch_geometric/visualization/influence.py,sha256=CWMvuNA_Nf1sfbJmQgn58yS4OFpeKXeZPe7kEuvkUBw,477
-pyg_nightly-2.7.0.dev20250326.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
-pyg_nightly-2.7.0.dev20250326.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-pyg_nightly-2.7.0.dev20250326.dist-info/METADATA,sha256=Kavjs9vEoxU7sQmFNxVUZioGBkdcRXGVkajH8b32yTk,63021
-pyg_nightly-2.7.0.dev20250326.dist-info/RECORD,,
+pyg_nightly-2.7.0.dev20250327.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
+pyg_nightly-2.7.0.dev20250327.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+pyg_nightly-2.7.0.dev20250327.dist-info/METADATA,sha256=QHeZLWk6bzSKdmuCxjDYsCxE6Y_ny6HnTIemg7Z4624,63021
+pyg_nightly-2.7.0.dev20250327.dist-info/RECORD,,

torch_geometric/__init__.py

@@ -31,7 +31,7 @@ from .lazy_loader import LazyLoader
 contrib = LazyLoader('contrib', globals(), 'torch_geometric.contrib')
 graphgym = LazyLoader('graphgym', globals(), 'torch_geometric.graphgym')
 
-__version__ = '2.7.0.dev20250326'
+__version__ = '2.7.0.dev20250327'
 
 __all__ = [
     'Index',

torch_geometric/utils/_negative_sampling.py

@@ -12,7 +12,7 @@ from torch_geometric.utils.num_nodes import maybe_num_nodes
 def negative_sampling(
     edge_index: Tensor,
     num_nodes: Optional[Union[int, Tuple[int, int]]] = None,
-    num_neg_samples: Optional[int] = None,
+    num_neg_samples: Optional[Union[int, float]] = None,
     method: str = "sparse",
     force_undirected: bool = False,
 ) -> Tensor:
@@ -25,10 +25,12 @@ def negative_sampling(
             If given as a tuple, then :obj:`edge_index` is interpreted as a
             bipartite graph with shape :obj:`(num_src_nodes, num_dst_nodes)`.
             (default: :obj:`None`)
-        num_neg_samples (int, optional): The (approximate) number of negative
-            samples to return.
-            If set to :obj:`None`, will try to return a negative edge for every
-            positive edge. (default: :obj:`None`)
+        num_neg_samples (int or float, optional): The (approximate) number of
+            negative samples to return. If set to a floating-point value, it
+            represents the ratio of negative samples to generate based on the
+            number of positive edges. If set to :obj:`None`, will try to
+            return a negative edge for every positive edge.
+            (default: :obj:`None`)
         method (str, optional): The method to use for negative sampling,
             *i.e.* :obj:`"sparse"` or :obj:`"dense"`.
             This is a memory/runtime trade-off.
@@ -48,6 +50,11 @@ def negative_sampling(
         tensor([[3, 0, 0, 3],
                 [2, 3, 2, 1]])
 
+        >>> negative_sampling(edge_index, num_nodes=(3, 4),
+        ...                   num_neg_samples=0.5)  # 50% of positive edges
+        tensor([[0, 3],
+                [3, 0]])
+
         >>> # For bipartite graph
         >>> negative_sampling(edge_index, num_nodes=(3, 4))
         tensor([[0, 2, 2, 1],
@@ -74,6 +81,8 @@ def negative_sampling(
 
     if num_neg_samples is None:
         num_neg_samples = edge_index.size(1)
+    elif isinstance(num_neg_samples, float):
+        num_neg_samples = int(num_neg_samples * edge_index.size(1))
     if force_undirected:
         num_neg_samples = num_neg_samples // 2
 
@@ -117,7 +126,7 @@ def negative_sampling(
 def batched_negative_sampling(
     edge_index: Tensor,
     batch: Union[Tensor, Tuple[Tensor, Tensor]],
-    num_neg_samples: Optional[int] = None,
+    num_neg_samples: Optional[Union[int, float]] = None,
     method: str = "sparse",
     force_undirected: bool = False,
 ) -> Tensor:
@@ -131,9 +140,11 @@ def batched_negative_sampling(
             node to a specific example.
             If given as a tuple, then :obj:`edge_index` is interpreted as a
             bipartite graph connecting two different node types.
-        num_neg_samples (int, optional): The number of negative samples to
-            return. If set to :obj:`None`, will try to return a negative edge
-            for every positive edge. (default: :obj:`None`)
+        num_neg_samples (int or float, optional): The number of negative
+            samples to return. If set to :obj:`None`, will try to return a
+            negative edge for every positive edge. If float, it will generate
+            :obj:`num_neg_samples * num_edges` negative samples.
+            (default: :obj:`None`)
         method (str, optional): The method to use for negative sampling,
             *i.e.* :obj:`"sparse"` or :obj:`"dense"`.
             This is a memory/runtime trade-off.
@@ -157,6 +168,11 @@ def batched_negative_sampling(
         tensor([[3, 1, 3, 2, 7, 7, 6, 5],
                 [2, 0, 1, 1, 5, 6, 4, 4]])
 
+        >>> # Using float multiplier for negative samples
+        >>> batched_negative_sampling(edge_index, batch, num_neg_samples=1.5)
+        tensor([[3, 1, 3, 2, 7, 7, 6, 5, 2, 0, 1, 1],
+                [2, 0, 1, 1, 5, 6, 4, 4, 3, 2, 3, 0]])
+
         >>> # For bipartite graph
         >>> edge_index1 = torch.as_tensor([[0, 0, 1, 1], [0, 1, 2, 3]])
         >>> edge_index2 = edge_index1 + torch.tensor([[2], [4]])