LearnerFactory.php

<?php

namespace App\Application\ML;

use Rubix\ML\Learner;
use Rubix\ML\NeuralNet\ActivationFunctions\ReLU;
use Rubix\ML\NeuralNet\Layers\Activation;
use Rubix\ML\NeuralNet\Layers\Dense;
use Rubix\ML\Pipeline;
use Rubix\ML\Regressors\MLPRegressor;
use Rubix\ML\Transformers\NumericStringConverter;
use Rubix\ML\Transformers\ZScaleStandardizer;

/**
 * @see https://docs.rubixml.com/latest/transformers/z-scale-standardizer.html
 * @see https://docs.rubixml.com/latest/regressors/mlp-regressor.html
 */
class LearnerFactory
{
    public static function createLearner(): Learner
    {
        return new Pipeline([
            new NumericStringConverter(),
            new ZScaleStandardizer(),
        ], new MLPRegressor([
            new Dense(8),
            new Activation(new ReLU()),
            new Dense(8),
            new Activation(new ReLU()),
        ]));
    }
}