@@ -128,7 +128,7 @@ |
||
128 | 128 | } |
129 | 129 | |
130 | 130 | /** |
131 | - * @return mixed |
|
131 | + * @return integer |
|
132 | 132 | */ |
133 | 133 | public function count() |
134 | 134 | { |
@@ -142,6 +142,9 @@ |
||
142 | 142 | $this->addLayer(new Layer($nodes, Input::class)); |
143 | 143 | } |
144 | 144 | |
145 | + /** |
|
146 | + * @param ActivationFunction $defaultActivationFunction |
|
147 | + */ |
|
145 | 148 | private function addNeuronLayers(array $layers, ?ActivationFunction $defaultActivationFunction = null): void |
146 | 149 | { |
147 | 150 | foreach ($layers as $layer) { |