mirror of
https://github.com/Llewellynvdm/php-ml.git
synced 2024-11-22 21:15:10 +00:00
726cf4cddf
* travis: move coveralls here, decouple from package * composer: use PSR4 * phpunit: simpler config * travis: add ecs run * composer: add ecs dev * use standard vendor/bin directory for dependency bins, confuses with local bins and require gitignore handling * ecs: add PSR2 * [cs] PSR2 spacing fixes * [cs] PSR2 class name fix * [cs] PHP7 fixes - return semicolon spaces, old rand functions, typehints * [cs] fix less strict typehints * fix typehints to make tests pass * ecs: ignore typehint-less elements * [cs] standardize arrays * [cs] standardize docblock, remove unused comments * [cs] use self where possible * [cs] sort class elements, from public to private * [cs] do not use yoda (found less yoda-cases, than non-yoda) * space * [cs] do not assign in condition * [cs] use namespace imports if possible * [cs] use ::class over strings * [cs] fix defaults for arrays properties, properties and constants single spacing * cleanup ecs comments * [cs] use item per line in multi-items array * missing line * misc * rebase
62 lines
1.9 KiB
PHP
62 lines
1.9 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace tests\Phpml\Clustering;
|
|
|
|
use Phpml\Clustering\KMeans;
|
|
use PHPUnit\Framework\TestCase;
|
|
|
|
class KMeansTest extends TestCase
|
|
{
|
|
public function testKMeansSamplesClustering(): void
|
|
{
|
|
$samples = [[1, 1], [8, 7], [1, 2], [7, 8], [2, 1], [8, 9]];
|
|
|
|
$kmeans = new KMeans(2);
|
|
$clusters = $kmeans->cluster($samples);
|
|
|
|
$this->assertCount(2, $clusters);
|
|
|
|
foreach ($samples as $index => $sample) {
|
|
if (in_array($sample, $clusters[0]) || in_array($sample, $clusters[1])) {
|
|
unset($samples[$index]);
|
|
}
|
|
}
|
|
|
|
$this->assertCount(0, $samples);
|
|
}
|
|
|
|
public function testKMeansInitializationMethods(): void
|
|
{
|
|
$samples = [
|
|
[180, 155], [186, 159], [119, 185], [141, 147], [157, 158],
|
|
[176, 122], [194, 160], [113, 193], [190, 148], [152, 154],
|
|
[162, 146], [188, 144], [185, 124], [163, 114], [151, 140],
|
|
[175, 131], [186, 162], [181, 195], [147, 122], [143, 195],
|
|
[171, 119], [117, 165], [169, 121], [159, 160], [159, 112],
|
|
[115, 122], [149, 193], [156, 135], [118, 120], [139, 159],
|
|
[150, 115], [181, 136], [167, 162], [132, 115], [175, 165],
|
|
[110, 147], [175, 118], [113, 145], [130, 162], [195, 179],
|
|
[164, 111], [192, 114], [194, 149], [139, 113], [160, 168],
|
|
[162, 110], [174, 144], [137, 142], [197, 160], [147, 173],
|
|
];
|
|
|
|
$kmeans = new KMeans(4, KMeans::INIT_KMEANS_PLUS_PLUS);
|
|
$clusters = $kmeans->cluster($samples);
|
|
$this->assertCount(4, $clusters);
|
|
|
|
$kmeans = new KMeans(4, KMeans::INIT_RANDOM);
|
|
$clusters = $kmeans->cluster($samples);
|
|
$this->assertCount(4, $clusters);
|
|
}
|
|
|
|
/**
|
|
* @expectedException \Phpml\Exception\InvalidArgumentException
|
|
*/
|
|
public function testThrowExceptionOnInvalidClusterNumber(): void
|
|
{
|
|
new KMeans(0);
|
|
}
|
|
}
|