PowerStationReport.php

<?php

namespace App\Application\ML;

use App\Application\Path\AppPathResolver;
use App\Domain\FileNames;
use Rubix\ML\CrossValidation\Reports\AggregateReport;
use Rubix\ML\CrossValidation\Reports\ErrorAnalysis;
use Rubix\ML\Datasets\Labeled;
use Rubix\ML\Extractors\CSV;
use Rubix\ML\PersistentModel;
use Rubix\ML\Persisters\Filesystem;
use Rubix\ML\Report;

/**
 * @see https://docs.rubixml.com/latest/cross-validation.html
 * @see https://docs.rubixml.com/latest/cross-validation/reports/error-analysis.html
 */
readonly class PowerStationReport
{
    public function __construct(
        private AppPathResolver $appPathResolver,
    ) {
    }

    public function generateReport(): Report
    {
        $dataset = Labeled::fromIterator(new CSV(
            $this->appPathResolver->getResourcesPath(FileNames::TEST_SET),
            header: false,
        ))->transformLabels(fn ($value) => (float) $value);

        $estimator = PersistentModel::load(new Filesystem(
            $this->appPathResolver->getResourcesPath(FileNames::MODEL_FILENAME),
        ));

        $predictions = $estimator->predict($dataset);

        $report = new AggregateReport([
            new ErrorAnalysis(),
        ]);

        return $report->generate($predictions, $dataset->labels());
    }
}