Total Complexity | 6 |
Total Lines | 38 |
Duplicated Lines | 0 % |
Changes | 0 |
1 | <?php |
||
10 | class StratifiedRandomSplit extends RandomSplit |
||
11 | { |
||
12 | protected function splitDataset(Dataset $dataset, float $testSize): void |
||
13 | { |
||
14 | $datasets = $this->splitByTarget($dataset); |
||
15 | |||
16 | foreach ($datasets as $targetSet) { |
||
17 | parent::splitDataset($targetSet, $testSize); |
||
18 | } |
||
19 | } |
||
20 | |||
21 | /** |
||
22 | * @return Dataset[] |
||
23 | */ |
||
24 | private function splitByTarget(Dataset $dataset): array |
||
38 | } |
||
39 | |||
40 | private function createDatasets(array $uniqueTargets, array $split): array |
||
48 | } |
||
49 | } |
||
50 |