Fix to edge iterator.

Author: otiliastr

research/gam/gam/data/dataset.py

@@ -259,6 +259,12 @@ def __init__(self, src, tgt, weight=None):
       self.tgt = tgt
       self.weight = weight
 
+    def copy(self, src=None, tgt=None, weight=None):
+      src = src if src is not None else self.src
+      tgt = tgt if tgt is not None else self.tgt
+      weight = weight if weight is not None else self.weight
+      return GraphDataset.Edge(src, tgt, weight)
+
   def __init__(self,
                name,
                features,

research/gam/gam/trainer/trainer_classification.py

@@ -661,9 +661,11 @@ def edge_iterator(self, data, batch_size, labeling):
       edges = data.get_edges(
           src_labeled=True, tgt_labeled=True, label_must_match=True)
     elif labeling == 'lu':
-      edges = (
-          data.get_edges(src_labeled=True, tgt_labeled=False) +
-          data.get_edges(src_labeled=False, tgt_labeled=True))
+      edges_lu = data.get_edges(src_labeled=True, tgt_labeled=False)
+      edges_ul = data.get_edges(src_labeled=False, tgt_labeled=True)
+      # Reverse the edges of UL to be LU.
+      edges_ul = [e.copy(src=e.tgt, tgt=e.src) for e in edges_ul]
+      edges = edges_lu + edges_ul
     elif labeling == 'uu':
       edges = data.get_edges(src_labeled=False, tgt_labeled=False)
     else:

research/gam/gam/trainer/trainer_classification_gcn.py

@@ -689,9 +689,11 @@ def edge_iterator(self, data, batch_size, labeling):
       edges = data.get_edges(
           src_labeled=True, tgt_labeled=True, label_must_match=True)
     elif labeling == 'lu':
-      edges = (
-          data.get_edges(src_labeled=True, tgt_labeled=False) +
-          data.get_edges(src_labeled=False, tgt_labeled=True))
+      edges_lu = data.get_edges(src_labeled=True, tgt_labeled=False)
+      edges_ul = data.get_edges(src_labeled=False, tgt_labeled=True)
+      # Reverse the edges of UL to be LU.
+      edges_ul = [e.copy(src=e.tgt, tgt=e.src) for e in edges_ul]
+      edges = edges_lu + edges_ul
     elif labeling == 'uu':
       edges = data.get_edges(src_labeled=False, tgt_labeled=False)
     else: