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

Improved code [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2022-02-13 00:07:54 -08:00
parent 8374498e6c
commit 21ca5d3845
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/ivfkmeans.c
View File

@@ -218,6 +218,7 @@ SampleRows(IvfflatBuildState * buildstate)
BlockNumber totalblocks = RelationGetNumberOfBlocks(buildstate->heap); BlockNumber totalblocks = RelationGetNumberOfBlocks(buildstate->heap);
buildstate->rowstoskip = -1; buildstate->rowstoskip = -1;
buildstate->samples->length = 0;
BlockSampler_Init(&buildstate->bs, totalblocks, targsamples, random()); BlockSampler_Init(&buildstate->bs, totalblocks, targsamples, random());
@@ -285,9 +286,6 @@ MiniBatchKmeans(IvfflatBuildState * buildstate)
/* Can take a while, so ensure we can interrupt */ /* Can take a while, so ensure we can interrupt */
CHECK_FOR_INTERRUPTS(); CHECK_FOR_INTERRUPTS();
/* Reset samples */
buildstate->samples->length = 0;
/* Get b examples picked randomly from X */ /* Get b examples picked randomly from X */
SampleRows(buildstate); SampleRows(buildstate);
m = buildstate->samples; m = buildstate->samples;
@@ -330,7 +328,7 @@ MiniBatchKmeans(IvfflatBuildState * buildstate)
eta = 1.0 / v[d[j]]; eta = 1.0 / v[d[j]];
/* Take gradient step */ /* Take gradient step */
for (k = 0; k < buildstate->dimensions; k++) for (k = 0; k < c->dim; k++)
c->x[k] = (1 - eta) * c->x[k] + eta * x->x[k]; c->x[k] = (1 - eta) * c->x[k] + eta * x->x[k];
} }