node, edge features to vary across labels

torch_geometric/datasets/fake.py

@@ -72,24 +72,24 @@ def generate_data(self) -> Data:
 
         data = Data()
 
+        if self._num_classes > 0 and self.task == 'node':
+            data.y = torch.randint(self._num_classes, (num_nodes, ))
+        elif self._num_classes > 0 and self.task == 'graph':
+            data.y = torch.tensor([random.randint(0, self._num_classes - 1)])
+
         data.edge_index = get_edge_index(num_nodes, num_nodes, self.avg_degree,
                                          self.is_undirected, remove_loops=True)
 
         if self.num_channels > 0:
-            data.x = torch.randn(num_nodes, self.num_channels)
+            data.x = torch.randn(num_nodes, self.num_channels) + data.y
         else:
             data.num_nodes = num_nodes
 
         if self.edge_dim > 1:
-            data.edge_attr = torch.rand(data.num_edges, self.edge_dim)
+            data.edge_attr = torch.rand(data.num_edges, self.edge_dim) + data.y
         elif self.edge_dim == 1:
             data.edge_weight = torch.rand(data.num_edges)
 
-        if self._num_classes > 0 and self.task == 'node':
-            data.y = torch.randint(self._num_classes, (num_nodes, ))
-        elif self._num_classes > 0 and self.task == 'graph':
-            data.y = torch.tensor([random.randint(0, self._num_classes - 1)])
-
         return data