PredictCommand.php

<?php

namespace App\UI\CLI;

use App\Application\Path\AppPathResolver;
use App\Domain\FileNames;
use League\Csv\CannotInsertRecord;
use League\Csv\Exception;
use League\Csv\UnavailableStream;
use League\Csv\Writer;
use Rubix\ML\Datasets\Labeled;
use Rubix\ML\Extractors\CSV;
use Rubix\ML\PersistentModel;
use Rubix\ML\Persisters\Filesystem;
use Symfony\Component\Console\Attribute\AsCommand;
use Symfony\Component\Console\Command\Command;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Output\OutputInterface;
use Symfony\Component\Console\Style\SymfonyStyle;

#[AsCommand(
    name: 'app:predict',
)]
class PredictCommand extends Command
{
    public function __construct(
        private readonly AppPathResolver $appPathResolver,
    ) {
        parent::__construct();
    }

    /**
     * @throws UnavailableStream
     * @throws CannotInsertRecord
     * @throws Exception
     */
    protected function execute(InputInterface $input, OutputInterface $output): int
    {
        $io = new SymfonyStyle($input, $output);

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

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

        $writer = Writer::createFromPath($this->appPathResolver->getResourcesPath(
            FileNames::PREDICTIONS_FILENAME,
        ), 'w');

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

        foreach ($predictions as $prediction) {
            $writer->insertOne([$prediction]);
        }

        $io->success('Results saved to file');

        return Command::SUCCESS;
    }
}