Proyectos de Subversion Moodle

<?php

declare(strict_types=1);

namespace Phpml\CrossValidation;

use Phpml\Dataset\Dataset;

class RandomSplit extends Split

{

    protected function splitDataset(Dataset $dataset, float $testSize): void

    {

        $samples = $dataset->getSamples();

        $labels = $dataset->getTargets();

        $datasetSize = count($samples);

        $testCount = count($this->testSamples);

        for ($i = $datasetSize; $i > 0; --$i) {

            $key = mt_rand(0, $datasetSize - 1);

            $setName = (count($this->testSamples) - $testCount) / $datasetSize >= $testSize ? 'train' : 'test';

            $this->{$setName.'Samples'}[] = $samples[$key];

            $this->{$setName.'Labels'}[] = $labels[$key];

        }

    }

}