liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-07-13 08:06:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Updated comments [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2024-04-11 22:35:11 -07:00
parent cc4b01bd49
commit 06d90fdc76
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/ivfkmeans.c
View File

@@ -228,6 +228,7 @@ ComputeNewCenters(VectorArray samples, VectorArray aggCenters, VectorArray newCe
int numCenters = aggCenters->maxlen; int numCenters = aggCenters->maxlen;
int numSamples = samples->length; int numSamples = samples->length;
/* Reset sum and count */
for (int j = 0; j < numCenters; j++) for (int j = 0; j < numCenters; j++)
{ {
Vector *vec = (Vector *) VectorArrayGet(aggCenters, j); Vector *vec = (Vector *) VectorArrayGet(aggCenters, j);
@@ -268,7 +269,7 @@ ComputeNewCenters(VectorArray samples, VectorArray aggCenters, VectorArray newCe
for (int j = 0; j < numSamples; j++) for (int j = 0; j < numSamples; j++)
centerCounts[closestCenters[j]] += 1; centerCounts[closestCenters[j]] += 1;
/* Average centers */ /* Divide sum by count */
for (int j = 0; j < numCenters; j++) for (int j = 0; j < numCenters; j++)
{ {
Vector *vec = (Vector *) VectorArrayGet(aggCenters, j); Vector *vec = (Vector *) VectorArrayGet(aggCenters, j);