mirror of
https://github.com/Llewellynvdm/php-ml.git
synced 2024-11-25 14:27:34 +00:00
a348111e97
* tests: update to PHPUnit 6.0 with rector * fix namespaces on tests * composer + tests: use standard test namespace naming * update travis * resolve conflict * phpstan lvl 2 * phpstan lvl 3 * phpstan lvl 4 * phpstan lvl 5 * phpstan lvl 6 * phpstan lvl 7 * level max * resolve conflict * [cs] clean empty docs * composer: bump to PHPUnit 6.4 * cleanup * composer + travis: add phpstan * phpstan lvl 1 * composer: update dev deps * phpstan fixes * update Contributing with new tools * docs: link fixes, PHP version update * composer: drop php-cs-fixer, cs already handled by ecs * ecs: add old set rules * [cs] apply rest of rules
61 lines
2.0 KiB
PHP
61 lines
2.0 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace Phpml\Tests\CrossValidation;
|
|
|
|
use Phpml\CrossValidation\StratifiedRandomSplit;
|
|
use Phpml\Dataset\ArrayDataset;
|
|
use PHPUnit\Framework\TestCase;
|
|
|
|
class StratifiedRandomSplitTest extends TestCase
|
|
{
|
|
public function testDatasetStratifiedRandomSplitWithEvenDistribution(): void
|
|
{
|
|
$dataset = new ArrayDataset(
|
|
$samples = [[1], [2], [3], [4], [5], [6], [7], [8]],
|
|
$labels = ['a', 'a', 'a', 'a', 'b', 'b', 'b', 'b']
|
|
);
|
|
|
|
$split = new StratifiedRandomSplit($dataset, 0.5);
|
|
|
|
$this->assertEquals(2, $this->countSamplesByTarget($split->getTestLabels(), 'a'));
|
|
$this->assertEquals(2, $this->countSamplesByTarget($split->getTestLabels(), 'b'));
|
|
|
|
$split = new StratifiedRandomSplit($dataset, 0.25);
|
|
|
|
$this->assertEquals(1, $this->countSamplesByTarget($split->getTestLabels(), 'a'));
|
|
$this->assertEquals(1, $this->countSamplesByTarget($split->getTestLabels(), 'b'));
|
|
}
|
|
|
|
public function testDatasetStratifiedRandomSplitWithEvenDistributionAndNumericTargets(): void
|
|
{
|
|
$dataset = new ArrayDataset(
|
|
$samples = [[1], [2], [3], [4], [5], [6], [7], [8]],
|
|
$labels = [1, 2, 1, 2, 1, 2, 1, 2]
|
|
);
|
|
|
|
$split = new StratifiedRandomSplit($dataset, 0.5);
|
|
|
|
$this->assertEquals(2, $this->countSamplesByTarget($split->getTestLabels(), 1));
|
|
$this->assertEquals(2, $this->countSamplesByTarget($split->getTestLabels(), 2));
|
|
|
|
$split = new StratifiedRandomSplit($dataset, 0.25);
|
|
|
|
$this->assertEquals(1, $this->countSamplesByTarget($split->getTestLabels(), 1));
|
|
$this->assertEquals(1, $this->countSamplesByTarget($split->getTestLabels(), 2));
|
|
}
|
|
|
|
private function countSamplesByTarget($splitTargets, $countTarget): int
|
|
{
|
|
$count = 0;
|
|
foreach ($splitTargets as $target) {
|
|
if ($target === $countTarget) {
|
|
++$count;
|
|
}
|
|
}
|
|
|
|
return $count;
|
|
}
|
|
}
|