php-ml/tests/Phpml/Clustering/DBSCANTest.php

97 lines
2.4 KiB
PHP
Raw Normal View History

2016-04-30 22:47:44 +00:00
<?php
2016-04-30 22:56:43 +00:00
2016-11-20 21:53:17 +00:00
declare(strict_types=1);
2016-04-30 22:47:44 +00:00
namespace Phpml\Tests\Clustering;
2016-04-30 22:47:44 +00:00
use Phpml\Clustering\DBSCAN;
2017-02-03 11:58:25 +00:00
use PHPUnit\Framework\TestCase;
2016-04-30 22:47:44 +00:00
2017-02-03 11:58:25 +00:00
class DBSCANTest extends TestCase
2016-04-30 22:47:44 +00:00
{
public function testDBSCANSamplesClustering(): void
2016-04-30 22:47:44 +00:00
{
2016-04-30 22:56:43 +00:00
$samples = [[1, 1], [8, 7], [1, 2], [7, 8], [2, 1], [8, 9]];
2016-04-30 22:47:44 +00:00
$clustered = [
[[1, 1], [1, 2], [2, 1]],
2016-04-30 22:56:43 +00:00
[[8, 7], [7, 8], [8, 9]],
2016-04-30 22:47:44 +00:00
];
$dbscan = new DBSCAN($epsilon = 2, $minSamples = 3);
$this->assertEquals($clustered, $dbscan->cluster($samples));
2016-05-01 21:17:09 +00:00
$samples = [[1, 1], [6, 6], [1, -1], [5, 6], [-1, -1], [7, 8], [-1, 1], [7, 7]];
2016-04-30 22:47:44 +00:00
$clustered = [
2016-04-30 22:56:43 +00:00
[[1, 1], [1, -1], [-1, -1], [-1, 1]],
[[6, 6], [5, 6], [7, 8], [7, 7]],
2016-04-30 22:47:44 +00:00
];
$dbscan = new DBSCAN($epsilon = 3, $minSamples = 4);
$this->assertEquals($clustered, $dbscan->cluster($samples));
}
public function testDBSCANSamplesClusteringAssociative(): void
{
$samples = [
'a' => [1, 1],
'b' => [9, 9],
'c' => [1, 2],
'd' => [9, 8],
'e' => [7, 7],
'f' => [8, 7],
];
$clustered = [
[
'a' => [1, 1],
'c' => [1, 2],
],
[
'b' => [9, 9],
'd' => [9, 8],
'e' => [7, 7],
'f' => [8, 7],
],
];
$dbscan = new DBSCAN($epsilon = 3, $minSamples = 2);
$this->assertEquals($clustered, $dbscan->cluster($samples));
}
public function testClusterEpsilonSmall(): void
{
$samples = [[0], [1], [2]];
$clustered = [
];
$dbscan = new DBSCAN($epsilon = 0.5, $minSamples = 2);
$this->assertEquals($clustered, $dbscan->cluster($samples));
}
public function testClusterEpsilonBoundary(): void
{
$samples = [[0], [1], [2]];
$clustered = [
];
$dbscan = new DBSCAN($epsilon = 1.0, $minSamples = 2);
$this->assertEquals($clustered, $dbscan->cluster($samples));
}
public function testClusterEpsilonLarge(): void
{
$samples = [[0], [1], [2]];
$clustered = [
[[0], [1], [2]],
];
$dbscan = new DBSCAN($epsilon = 1.5, $minSamples = 2);
$this->assertEquals($clustered, $dbscan->cluster($samples));
}
2016-04-30 22:47:44 +00:00
}