php-ml/tests/CrossValidation/StratifiedRandomSplitTest.php

64 lines
2.0 KiB
PHP
Raw Normal View History

<?php
2016-11-20 21:53:17 +00:00
declare(strict_types=1);
namespace Phpml\Tests\CrossValidation;
use Phpml\CrossValidation\StratifiedRandomSplit;
use Phpml\Dataset\ArrayDataset;
2017-02-03 11:58:25 +00:00
use PHPUnit\Framework\TestCase;
2017-02-03 11:58:25 +00:00
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);
2018-10-28 06:44:52 +00:00
self::assertEquals(2, $this->countSamplesByTarget($split->getTestLabels(), 'a'));
self::assertEquals(2, $this->countSamplesByTarget($split->getTestLabels(), 'b'));
$split = new StratifiedRandomSplit($dataset, 0.25);
2018-10-28 06:44:52 +00:00
self::assertEquals(1, $this->countSamplesByTarget($split->getTestLabels(), 'a'));
self::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);
2018-10-28 06:44:52 +00:00
self::assertEquals(2, $this->countSamplesByTarget($split->getTestLabels(), 1));
self::assertEquals(2, $this->countSamplesByTarget($split->getTestLabels(), 2));
$split = new StratifiedRandomSplit($dataset, 0.25);
2018-10-28 06:44:52 +00:00
self::assertEquals(1, $this->countSamplesByTarget($split->getTestLabels(), 1));
self::assertEquals(1, $this->countSamplesByTarget($split->getTestLabels(), 2));
}
2018-10-28 06:44:52 +00:00
/**
* @param string|int $countTarget
*/
private function countSamplesByTarget(array $splitTargets, $countTarget): int
{
$count = 0;
foreach ($splitTargets as $target) {
if ($target === $countTarget) {
++$count;
}
}
return $count;
}
}