PowerStationTrainer.php

<?php

namespace App\Application\ML;

use App\Application\Path\AppPathResolver;
use App\Domain\FileNames;
use Rubix\ML\Datasets\Labeled;
use Rubix\ML\Extractors\CSV;
use Rubix\ML\PersistentModel;
use Rubix\ML\Persisters\Filesystem;

readonly class PowerStationTrainer
{
    public function __construct(
        private AppPathResolver $appPathResolver,
    ) {
    }

    public function train(bool $history = false): void
    {
        $dataset = Labeled::fromIterator(new CSV(
            $this->appPathResolver->getResourcesPath(FileNames::TRAINING_SET),
            header: false,
        ))->transformLabels(fn ($value) => (float) $value);

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

        $estimator->train($dataset);
        $estimator->save();
    }
}