From 47ac8186010bcd14ce14493f02075962d6b359d5 Mon Sep 17 00:00:00 2001 From: Matthias Fey Date: Tue, 21 Jan 2025 10:34:49 +0100 Subject: [PATCH] Fix index-out-of-bounds issue in `LinkPredNDCG(weighted=True)` (#9963) --- torch_geometric/metrics/link_pred.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_geometric/metrics/link_pred.py b/torch_geometric/metrics/link_pred.py index b9ad50a642a5..95e274dcd619 100644 --- a/torch_geometric/metrics/link_pred.py +++ b/torch_geometric/metrics/link_pred.py @@ -492,7 +492,7 @@ def _compute(self, data: LinkPredMetricData) -> Tensor: self.discount, self.discount.new_full((1, ), fill_value=float('inf')), ]) - discount = discount[pos.clamp(max=self.k + 1)] + discount = discount[pos.clamp(max=self.k)] idcg = scatter( # Apply discount and aggregate: data.edge_label_weight / discount,