Merge pull request #104 from STayinloves/master

Fix incorrect relation type measurement
uma-pi1 · May 23, 2020 · 1ec6a1d · 1ec6a1d
2 parents 71c96ab + 5dcceb4
commit 1ec6a1d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/kge/indexing.py b/kge/indexing.py
@@ -99,7 +99,9 @@ def _get_relation_types(dataset,):
         (dataset.index("train_po_to_s"), 0),
     ]:
         for prefix, labels in index.items():
-            relation_stats[prefix[p], 0 + p * 2] = labels.float().sum()
+            relation_stats[prefix[p], 0 + p * 2] = (
+                relation_stats[prefix[p], 0 + p * 2] + len(labels)
+            )
             relation_stats[prefix[p], 1 + p * 2] = (
                 relation_stats[prefix[p], 1 + p * 2] + 1.0
             )