PowerStationTester.php

<?php

namespace App\Application\ML;

use App\Application\Path\AppPathResolver;
use App\Domain\FileNames;
use Rubix\ML\CrossValidation\KFold;
use Rubix\ML\CrossValidation\Metrics\MeanSquaredError;
use Rubix\ML\Datasets\Labeled;
use Rubix\ML\Extractors\CSV;

/**
 * @see https://docs.rubixml.com/latest/cross-validation.html#validators
 * @see https://docs.rubixml.com/latest/cross-validation/metrics/mean-squared-error.html
 */
readonly class PowerStationTester
{
    public function __construct(
        private AppPathResolver $appPathResolver,
    ) {
    }

    public function test(int $foldsNumber = 5): float
    {
        $dataset = Labeled::fromIterator(new CSV(
            $this->appPathResolver->getResourcesPath(FileNames::VALID_DATA),
            header: true,
        ))->transformLabels(fn ($value) => (float) $value);

        $estimator = LearnerFactory::createLearner();
        $validator = new KFold($foldsNumber);

        return $validator->test($estimator, $dataset, new MeanSquaredError());
    }
}