diff options
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.java | 6 |
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++) { |