php-ml/tests/Clustering/FuzzyCMeansTest.php
2018-10-28 07:44:52 +01:00

65 lines
1.7 KiB
PHP

<?php
declare(strict_types=1);
namespace Phpml\Tests\Clustering;
use Phpml\Clustering\FuzzyCMeans;
use Phpml\Exception\InvalidArgumentException;
use PHPUnit\Framework\TestCase;
class FuzzyCMeansTest extends TestCase
{
public function testFCMSamplesClustering(): void
{
$samples = [[1, 1], [8, 7], [1, 2], [7, 8], [2, 1], [8, 9]];
$fcm = new FuzzyCMeans(2);
$clusters = $fcm->cluster($samples);
self::assertCount(2, $clusters);
foreach ($samples as $index => $sample) {
if (in_array($sample, $clusters[0], true) || in_array($sample, $clusters[1], true)) {
unset($samples[$index]);
}
}
self::assertCount(0, $samples);
}
public function testMembershipMatrix(): void
{
$fcm = new FuzzyCMeans(2);
$fcm->cluster([[1, 1], [8, 7], [1, 2], [7, 8], [2, 1], [8, 9]]);
$clusterCount = 2;
$sampleCount = 6;
$matrix = $fcm->getMembershipMatrix();
self::assertCount($clusterCount, $matrix);
foreach ($matrix as $row) {
self::assertCount($sampleCount, $row);
}
// Transpose of the matrix
array_unshift($matrix, null);
$matrix = array_map(...$matrix);
// All column totals should be equal to 1 (100% membership)
foreach ($matrix as $col) {
self::assertEquals(1, array_sum($col));
}
}
/**
* @dataProvider invalidClusterNumberProvider
*/
public function testInvalidClusterNumber(int $clusters): void
{
$this->expectException(InvalidArgumentException::class);
new FuzzyCMeans($clusters);
}
public function invalidClusterNumberProvider(): array
{
return [[0], [-1]];
}
}