AMRI in the final test eval, docstring for a dataset

pykeen · Feb 22, 2022 · 0211b76 · 0211b76
1 parent ec2e14f
commit 0211b76
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/dataset.py b/dataset.py
@@ -14,7 +14,7 @@
 
 
 class InductiveLPDataset(DisjointInductivePathDataset):
-    """An inductive link prediction dataset for the `KG Course Competition <https://github.com/migalkin/kgcourse2021>`_."""
+    """An inductive link prediction dataset for the ILPC 2022 Challenge."""
 
     def __init__(self, size: Size = "small", **kwargs):
         """Initialize the inductive link prediction dataset.

diff --git a/main.py b/main.py
@@ -183,7 +183,8 @@ def main(
         f"Test Arithmetic Mean Rank {results_dict['arithmetic_mean_rank']['both']['realistic']:.5f}"
     )
     print(
-        f"Test Adjusted Arithmetic Mean Rank {results_dict['adjusted_arithmetic_mean_rank']['both']['realistic']:.5f}"
+        f"Test Adjusted Arithmetic Mean Rank Index"
+        f" {results_dict['adjusted_arithmetic_mean_rank_index']['both']['realistic']:.5f}"
     )
 
     # you can also log the final results to wandb if you want