octoleo/php-ml
1
0
Fork 0
mirror of https://github.com/Llewellynvdm/php-ml.git synced 2025-01-10 00:37:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

test kmeans init methods

Browse Source
This commit is contained in:
Arkadiusz Kondas 2016-05-01 23:44:04 +02:00
parent 7572304d50
commit bb9e1aa4f0
1 changed files with 15 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
tests/Phpml/Clustering/KMeansTest.php
View File

@ -25,33 +25,27 @@ class KMeansTest extends \PHPUnit_Framework_TestCase
$this->assertEquals(0, count($samples)); $this->assertEquals(0, count($samples));
} }
public function testKMeansMoreSamplesClustering() public function testKMeansInitializationMethods()
{ {
$samples = [ $samples = [
[80, 55], [86, 59], [19, 85], [41, 47], [57, 58], [180, 155], [186, 159], [119, 185], [141, 147], [157, 158],
[76, 22], [94, 60], [13, 93], [90, 48], [52, 54], [176, 122], [194, 160], [113, 193], [190, 148], [152, 154],
[62, 46], [88, 44], [85, 24], [63, 14], [51, 40], [162, 146], [188, 144], [185, 124], [163, 114], [151, 140],
[75, 31], [86, 62], [81, 95], [47, 22], [43, 95], [175, 131], [186, 162], [181, 195], [147, 122], [143, 195],
[71, 19], [17, 65], [69, 21], [59, 60], [59, 12], [171, 119], [117, 165], [169, 121], [159, 160], [159, 112],
[15, 22], [49, 93], [56, 35], [18, 20], [39, 59], [115, 122], [149, 193], [156, 135], [118, 120], [139, 159],
[50, 15], [81, 36], [67, 62], [32, 15], [75, 65], [150, 115], [181, 136], [167, 162], [132, 115], [175, 165],
[10, 47], [75, 18], [13, 45], [30, 62], [95, 79], [110, 147], [175, 118], [113, 145], [130, 162], [195, 179],
[64, 11], [92, 14], [94, 49], [39, 13], [60, 68], [164, 111], [192, 114], [194, 149], [139, 113], [160, 168],
[62, 10], [74, 44], [37, 42], [97, 60], [47, 73], [162, 110], [174, 144], [137, 142], [197, 160], [147, 173],
]; ];
$kmeans = new KMeans(4); $kmeans = new KMeans(4, KMeans::INIT_KMEANS_PLUS_PLUS);
$clusters = $kmeans->cluster($samples); $clusters = $kmeans->cluster($samples);
$this->assertEquals(4, count($clusters)); $this->assertEquals(4, count($clusters));
foreach ($samples as $index => $sample) { $kmeans = new KMeans(4, KMeans::INIT_RANDOM);
for ($i = 0; $i < 4; ++$i) { $clusters = $kmeans->cluster($samples);
if (in_array($sample, $clusters[$i])) { $this->assertEquals(4, count($clusters));
unset($samples[$index]);
}
}
}
$this->assertEquals(0, count($samples));
} }
} }