Skip to content

Commit 234a0a8

Browse files
committed
bad assert and hardcoding to 8 for target size for now for parent centroid clustering
1 parent 1b4f69c commit 234a0a8

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

server/src/main/java/org/elasticsearch/index/codec/vectors/DefaultIVFVectorsWriter.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -381,7 +381,7 @@ public DocIndexIterator iterator() {
381381

382382
HierarchicalKMeans hierarchicalKMeans = new HierarchicalKMeans(fieldInfo.getVectorDimension());
383383
// TODO: evaluate further whether 512 is better than other sizes like 32 or sqrt(centroids.length)
384-
return hierarchicalKMeans.cluster(centroidsAsFVV, 512); // centroids.length / (int) Math.sqrt(centroids.length));
384+
return hierarchicalKMeans.cluster(centroidsAsFVV, 8); // centroids.length / (int) Math.sqrt(centroids.length));
385385
}
386386

387387
static int[][] buildCentroidAssignments(int centroidCount, int[] assignments, int[] soarAssignments) {

server/src/main/java/org/elasticsearch/index/codec/vectors/IVFVectorsWriter.java

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,6 @@ private void mergeOneFieldIVF(FieldInfo fieldInfo, MergeState mergeState) throws
317317
numCentroids = centroidAssignments.numCentroids();
318318
numClusters = centroidAssignments.centroids().length;
319319
assignmentsByCluster = centroidAssignments.assignmentsByCluster();
320-
assert centroidAssignments.clusterToCentroidMap() != null;
321320
clusterToCentroidMap = centroidAssignments.clusterToCentroidMap();
322321
success = true;
323322
} finally {

0 commit comments

Comments
 (0)