Fix wront relation type measurement

uma-pi1 · May 23, 2020 · 5dcceb4 · 5dcceb4
1 parent 71c96ab
commit 5dcceb4
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
             )