kmeans()

static uint32_t kmeans ( POINT4D *  objs, uint32_t *  clusters, uint32_t  n, POINT4D *  centers, double *  radii, uint32_t  min_k, double  max_radius  )

static

Definition at line 276 of file lwkmeans.c.

{
        uint8_t converged = LW_FALSE;
        uint32_t cur_k = min_k;
 
        kmeans_init(objs, n, centers, cur_k);
        /* One iteration of kmeans needs to happen without shortcuts to fully initialize structures */
        update_r(objs, clusters, n, centers, radii, cur_k);
        update_means(objs, clusters, n, centers, cur_k);
        for (uint32_t t = 0; t < KMEANS_MAX_ITERATIONS; t++)
        {
                /* Standard KMeans loop */
                for (uint32_t i = 0; i < KMEANS_MAX_ITERATIONS; i++)
                {
                        LW_ON_INTERRUPT(break);
                        converged = update_r(objs, clusters, n, centers, radii, cur_k);
                        if (converged)
                                break;
                        update_means(objs, clusters, n, centers, cur_k);
                }
                if (!converged || !max_radius)
                        break;
 
                /* XMeans-inspired improve_structure pass to split clusters bigger than limit into 2 */
                uint32_t new_k = improve_structure(objs, clusters, n, centers, radii, cur_k, max_radius);
                if (new_k == cur_k)
                        break;
                cur_k = new_k;
        }
 
        if (!converged)
        {
                lwerror("%s did not converge after %d iterations", __func__, KMEANS_MAX_ITERATIONS);
                return 0;
        }
        return cur_k;
}

References improve_structure(), kmeans_init(), KMEANS_MAX_ITERATIONS, LW_FALSE, LW_ON_INTERRUPT, lwerror(), update_means(), and update_r().

Referenced by improve_structure(), and lwgeom_cluster_kmeans().

Here is the call graph for this function:

Here is the caller graph for this function: