Phpml\NeuralNetwork\Layer

Complexity

Total Complexity

7

Size/Duplication

Total Lines	56
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	2

Importance

Changes	2
Bugs	0	Features	2

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	10	3
A	createNode()	0	8	2
A	addNode()	0	4	1
A	getNodes()	0	4	1

<?php

declare (strict_types = 1);

namespace Phpml\NeuralNetwork;

use Phpml\Exception\InvalidArgumentException;
use Phpml\NeuralNetwork\Node\Neuron;

class Layer
{
    /**
     * @var Node[]
     */
    private $nodes = [];

    /**
     * @param int                     $nodesNumber
     * @param string                  $nodeClass
     * @param ActivationFunction|null $activationFunction
     *
     * @throws InvalidArgumentException
     */
    public function __construct(int $nodesNumber = 0, string $nodeClass = Neuron::class, ActivationFunction $activationFunction = null)
    {
        if (!in_array(Node::class, class_implements($nodeClass))) {
            throw InvalidArgumentException::invalidLayerNodeClass();
        }

        for ($i = 0; $i < $nodesNumber; ++$i) {
            $this->nodes[] = $this->createNode($nodeClass, $activationFunction);
        }
    }

    /**
     * @param string                  $nodeClass
     * @param ActivationFunction|null $activationFunction
     *
     * @return Neuron
     */
    private function createNode(string $nodeClass, ActivationFunction $activationFunction = null)
    {
        if (Neuron::class == $nodeClass) {
            return new Neuron($activationFunction);
        }

        return new $nodeClass();
    }

    /**
     * @param Node $node
     */
    public function addNode(Node $node)
    {
        $this->nodes[] = $node;
    }

    /**
     * @return Node[]
     */
    public function getNodes()
    {
        return $this->nodes;
    }
}