kmeans()

static int kmeans ( POINT2D **  objs, int *  clusters, uint32_t  n, POINT2D **  centers, uint32_t  k  )

static

Definition at line 97 of file lwkmeans.c.

{
        uint32_t i = 0;
        int* clusters_last;
        int converged = LW_FALSE;
        size_t clusters_sz = sizeof(int) * n;
        uint32_t* weights;
 
        weights = lwalloc(sizeof(int) * k);
 
        /* previous cluster state array */
        clusters_last = lwalloc(clusters_sz);
 
        for (i = 0; i < KMEANS_MAX_ITERATIONS && !converged; i++)
        {
                LW_ON_INTERRUPT(break);
 
                /* store the previous state of the clustering */
                memcpy(clusters_last, clusters, clusters_sz);
 
                update_r(objs, clusters, n, centers, k);
                update_means(objs, clusters, n, centers, weights, k);
 
                /* if all the cluster numbers are unchanged, we are at a stable solution */
                converged = memcmp(clusters_last, clusters, clusters_sz) == 0;
        }
 
        lwfree(clusters_last);
        lwfree(weights);
        if (!converged)
                lwerror("%s did not converge after %d iterations", __func__, i);
        return converged;
}

References KMEANS_MAX_ITERATIONS, LW_FALSE, LW_ON_INTERRUPT, lwalloc(), lwerror(), lwfree(), update_means(), and update_r().

Referenced by lwgeom_cluster_2d_kmeans().

Here is the call graph for this function:

Here is the caller graph for this function: