aboutsummaryrefslogtreecommitdiffstats
path: root/container-search/src/main/java/com/yahoo/search/ranking/ReciprocalRankNormalizer.java
diff options
context:
space:
mode:
Diffstat (limited to 'container-search/src/main/java/com/yahoo/search/ranking/ReciprocalRankNormalizer.java')
-rw-r--r--container-search/src/main/java/com/yahoo/search/ranking/ReciprocalRankNormalizer.java6
1 files changed, 4 insertions, 2 deletions
diff --git a/container-search/src/main/java/com/yahoo/search/ranking/ReciprocalRankNormalizer.java b/container-search/src/main/java/com/yahoo/search/ranking/ReciprocalRankNormalizer.java
index 7862e54c32e..d3cdfc4bc78 100644
--- a/container-search/src/main/java/com/yahoo/search/ranking/ReciprocalRankNormalizer.java
+++ b/container-search/src/main/java/com/yahoo/search/ranking/ReciprocalRankNormalizer.java
@@ -9,7 +9,7 @@ class ReciprocalRankNormalizer extends Normalizer {
ReciprocalRankNormalizer(String name, String input, int maxSize, double k) {
super(name, input, maxSize);
- this.k = k;
+ this.k = k;
}
static record IdxScore(int index, double score) {}
@@ -18,7 +18,9 @@ class ReciprocalRankNormalizer extends Normalizer {
if (size < 1) return;
IdxScore[] temp = new IdxScore[size];
for (int i = 0; i < size; i++) {
- temp[i] = new IdxScore(i, data[i]);
+ double val = data[i];
+ if (Double.isNaN(val)) val = Double.NEGATIVE_INFINITY;
+ temp[i] = new IdxScore(i, val);
}
Arrays.sort(temp, (a, b) -> Double.compare(b.score, a.score));
for (int i = 0; i < size; i++) {