update

pyg-team · Jan 14, 2025 · 4cc4411 · 4cc4411
1 parent 302e8c6
commit 4cc4411
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/torch_geometric/metrics/link_pred.py b/torch_geometric/metrics/link_pred.py
@@ -164,7 +164,10 @@ class LinkPredMetricCollection(torch.nn.ModuleDict):
         ])
 
         metrics.update(pred_index_mat, edge_label_index)
-        print(metrics.compute())
+        out = metrics.compute()
+        metrics.reset()
+
+        print(out)
         >>> {'LinkPredMAP@10': tensor(0.375),
         ...  'LinkPredPrecision@100': tensor(0.127),
         ...  'LinkPredRecall@50': tensor(0.483)}