Proyectos de Subversion Moodle

Rev

| Ultima modificación | Ver Log |

Rev Autor Línea Nro. Línea
1 efrain 1
<?php
2
 
3
declare(strict_types=1);
4
 
5
namespace Phpml\NeuralNetwork;
6
 
7
use Phpml\Exception\InvalidArgumentException;
8
use Phpml\NeuralNetwork\Node\Neuron;
9
 
10
class Layer
11
{
12
    /**
13
     * @var Node[]
14
     */
15
    private $nodes = [];
16
 
17
    /**
18
     * @throws InvalidArgumentException
19
     */
20
    public function __construct(int $nodesNumber = 0, string $nodeClass = Neuron::class, ?ActivationFunction $activationFunction = null)
21
    {
22
        if (!in_array(Node::class, class_implements($nodeClass), true)) {
23
            throw new InvalidArgumentException('Layer node class must implement Node interface');
24
        }
25
 
26
        for ($i = 0; $i < $nodesNumber; ++$i) {
27
            $this->nodes[] = $this->createNode($nodeClass, $activationFunction);
28
        }
29
    }
30
 
31
    public function addNode(Node $node): void
32
    {
33
        $this->nodes[] = $node;
34
    }
35
 
36
    /**
37
     * @return Node[]
38
     */
39
    public function getNodes(): array
40
    {
41
        return $this->nodes;
42
    }
43
 
44
    private function createNode(string $nodeClass, ?ActivationFunction $activationFunction = null): Node
45
    {
46
        if ($nodeClass === Neuron::class) {
47
            return new Neuron($activationFunction);
48
        }
49
 
50
        return new $nodeClass();
51
    }
52
}