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

Added inertia

Browse Source
This commit is contained in:
Andrew Kane
2022-02-12 21:23:53 -08:00
parent 71d2908be1
commit fa5e90585d
2 changed files with 19 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/ivfbuild.c
View File

@@ -170,6 +170,10 @@ BuildCallback(Relation index, CALLBACK_ITEM_POINTER, Datum *values,
} }
} }
#ifdef IVFFLAT_KMEANS_DEBUG
buildstate->inertia += minDistance;
#endif
/* Create a virtual tuple */ /* Create a virtual tuple */
ExecClearTuple(slot); ExecClearTuple(slot);
slot->tts_values[0] = Int32GetDatum(closestCenter); slot->tts_values[0] = Int32GetDatum(closestCenter);
@@ -350,6 +354,10 @@ InitBuildState(IvfflatBuildState * buildstate, Relation heap, Relation index, In
/* Reuse for each tuple */ /* Reuse for each tuple */
buildstate->normvec = InitVector(buildstate->dimensions); buildstate->normvec = InitVector(buildstate->dimensions);
#ifdef IVFFLAT_KMEANS_DEBUG
buildstate->inertia = 0;
#endif
} }
/* /*
@@ -502,8 +510,14 @@ CreateEntryPages(IvfflatBuildState * buildstate, ForkNumber forkNum)
#endif #endif
} }
/* Sort and insert */ /* Sort */
tuplesort_performsort(buildstate->sortstate); tuplesort_performsort(buildstate->sortstate);
#ifdef IVFFLAT_KMEANS_DEBUG
elog(INFO, "inertia: %.3e", buildstate->inertia);
#endif
/* Insert */
InsertTuples(buildstate->index, buildstate, forkNum); InsertTuples(buildstate->index, buildstate, forkNum);
tuplesort_end(buildstate->sortstate); tuplesort_end(buildstate->sortstate);
} }

4
src/ivfflat.h
View File

@@ -120,6 +120,10 @@ typedef struct IvfflatBuildState
ListInfo *listInfo; ListInfo *listInfo;
Vector *normvec; Vector *normvec;
#ifdef IVFFLAT_KMEANS_DEBUG
double inertia;
#endif
/* Sampling */ /* Sampling */
BlockSamplerData bs; BlockSamplerData bs;
ReservoirStateData rstate; ReservoirStateData rstate;