Fixing scoring normalization for search spanning multiple segments

Ishan Chattopadhyaya · Ishan Chattopadhyaya · commit e4e1b15cf499 · 2025-03-10T20:46:52.000+05:30
diff --git a/lucene/sandbox/src/java/org/apache/lucene/sandbox/vectorsearch/PerLeafCuVSKnnCollector.java b/lucene/sandbox/src/java/org/apache/lucene/sandbox/vectorsearch/PerLeafCuVSKnnCollector.java
@@ -71,7 +71,7 @@ public int k() {
   @Override
   @SuppressWarnings("cast")
   public boolean collect(int docId, float similarity) {
-    scoreDocs.add(new ScoreDoc(docId, 1f / (float) (similarity)));
+    scoreDocs.add(new ScoreDoc(docId, similarity));
     return true;
   }
 

Original file line number	Diff line number	Diff line change
`@@ -71,7 +71,7 @@ public int k() {`
`71`	`71`	`@Override`
`72`	`72`	`@SuppressWarnings("cast")`
`73`	`73`	`public boolean collect(int docId, float similarity) {`
`74`		`- scoreDocs.add(new ScoreDoc(docId, 1f / (float) (similarity)));`
	`74`	`+ scoreDocs.add(new ScoreDoc(docId, similarity));`
`75`	`75`	`return true;`
`76`	`76`	`}`
`77`	`77`