Kmeans\RandomInitialization::initializeClusters()

Complexity

Conditions	4
Paths	3

Size

Total Lines	16
Code Lines	6

Duplication

Lines	0
Ratio	0 %

Importance

Changes	1
Bugs	0	Features	0

<?php
    public function initializeClusters(PointCollectionInterface $points, int $nbClusters): ClusterCollectionInterface
    {
        // validate cluster count
        if ($nbClusters < 1) {
            throw new \InvalidArgumentException("Invalid cluster count: {$nbClusters}");
        }

        $clusters = new ClusterCollection($points->getSpace());

        // initialize N clusters with a random point within space boundaries
        for ($n = 0; $n < $nbClusters; $n++) {
            // assign all points to the first cluster only
            $clusters->attach(new Cluster($this->getRandomPoint($points), $n == 0 ? $points : null));
        }

        return $clusters;
    }