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

Improved code [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2022-02-15 18:32:13 -08:00
parent 2ee510aa67
commit 50349ed4f5
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/ivfkmeans.c
View File

@@ -252,7 +252,8 @@ static void
MiniBatchKmeans(IvfflatBuildState * buildstate) MiniBatchKmeans(IvfflatBuildState * buildstate)
{ {
VectorArray centers = buildstate->centers; VectorArray centers = buildstate->centers;
int b = buildstate->samples->maxlen; VectorArray m = buildstate->samples;
int b = m->maxlen;
int t = 20; int t = 20;
double distance; double distance;
double minDistance; double minDistance;
@@ -260,7 +261,6 @@ MiniBatchKmeans(IvfflatBuildState * buildstate)
int i; int i;
int j; int j;
int k; int k;
VectorArray m;
Vector *c; Vector *c;
Vector *x; Vector *x;
int *v; int *v;
@@ -288,7 +288,6 @@ MiniBatchKmeans(IvfflatBuildState * buildstate)
/* Get b examples picked randomly from X */ /* Get b examples picked randomly from X */
SampleRows(buildstate); SampleRows(buildstate);
m = buildstate->samples;
/* Cache nearest center to x */ /* Cache nearest center to x */
for (j = 0; j < m->length; j++) for (j = 0; j < m->length; j++)