Skip to content

Commit f6bc295

Browse files
committed
simplifying desired clusters
1 parent 812cdf9 commit f6bc295

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

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

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -65,10 +65,7 @@ CentroidAssignmentScorer calculateAndWriteCentroids(
6565
}
6666
// calculate the centroids
6767
int maxNumClusters = ((floatVectorValues.size() - 1) / vectorPerCluster) + 1;
68-
int desiredClusters = (int) Math.max(maxNumClusters / 16.0, Math.max(Math.sqrt(floatVectorValues.size()), maxNumClusters));
69-
if (floatVectorValues.size() / desiredClusters > vectorPerCluster) {
70-
desiredClusters = ((floatVectorValues.size() - 1) / vectorPerCluster) + 1;
71-
}
68+
int desiredClusters = (int) Math.max(Math.sqrt(floatVectorValues.size()), maxNumClusters);
7269
final KMeans.Results kMeans = KMeans.cluster(
7370
floatVectorValues,
7471
desiredClusters,

0 commit comments

Comments
 (0)