php-ml/src/Phpml/NeuralNetwork/Node/Neuron.php

66 lines
1.3 KiB
PHP

<?php
declare(strict_types=1);
namespace Phpml\NeuralNetwork\Node;
use Phpml\NeuralNetwork\ActivationFunction;
use Phpml\NeuralNetwork\ActivationFunction\Sigmoid;
use Phpml\NeuralNetwork\Node;
use Phpml\NeuralNetwork\Node\Neuron\Synapse;
class Neuron implements Node
{
/**
* @var Synapse[]
*/
protected $synapses = [];
/**
* @var ActivationFunction
*/
protected $activationFunction;
/**
* @var float
*/
protected $output = 0.0;
public function __construct(?ActivationFunction $activationFunction = null)
{
$this->activationFunction = $activationFunction ?: new Sigmoid();
}
public function addSynapse(Synapse $synapse): void
{
$this->synapses[] = $synapse;
}
/**
* @return Synapse[]
*/
public function getSynapses()
{
return $this->synapses;
}
public function getOutput(): float
{
if ($this->output === 0.0) {
$sum = 0.0;
foreach ($this->synapses as $synapse) {
$sum += $synapse->getOutput();
}
$this->output = $this->activationFunction->compute($sum);
}
return $this->output;
}
public function reset(): void
{
$this->output = 0.0;
}
}