Make ann use mutable object instead of numbers directly

Autapomorph · Nov 26, 2020 · 46a2a36 · 46a2a36
1 parent b9e5544
commit 46a2a36
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/src/ai/ann/Network.js b/src/ai/ann/Network.js
@@ -41,7 +41,7 @@ function Network(layerDescriptors, learningRate = 0.5) {
         network.layers[l].neurons.forEach((neuron, i) => {
           neuron.error = sum(
             network.layers[l + 1].neurons.map(function (n) {
-              return n.weights[i] * n.delta;
+              return n.weights[i].value * n.delta;
             }),
           );
 

diff --git a/src/ai/ann/Neuron.js b/src/ai/ann/Neuron.js
@@ -2,7 +2,7 @@ const { dot, activations } = require('./math');
 const { times } = require('lodash');
 
 function Neuron(inputQuantity = 1, type = 'linear') {
-  let weights = times(inputQuantity + 1, Math.random);
+  let weights = times(inputQuantity + 1, () => ({ value: Math.random() }));
   let neuron = {
     weights,
     predict,
@@ -13,12 +13,15 @@ function Neuron(inputQuantity = 1, type = 'linear') {
   };
 
   function _predict(inputs) {
-    return dot(neuron.weights, inputs);
+    return dot(
+      neuron.weights.map(w => w.value),
+      inputs,
+    );
   }
 
   function predict(inputs) {
     inputs = [...inputs, -1];
-    while (inputs.length > neuron.weights.length) neuron.weights.push(Math.random());
+    while (inputs.length > neuron.weights.length) neuron.weights.push({ value: Math.random() });
     neuron.inputs = inputs;
     neuron.output = activations[type].function(_predict(inputs));
 
@@ -27,7 +30,7 @@ function Neuron(inputQuantity = 1, type = 'linear') {
 
   function adjust(delta) {
     for (let i = 0; i < neuron.weights.length; i++) {
-      neuron.weights[i] += delta * neuron.inputs[i];
+      neuron.weights[i].value += delta * neuron.inputs[i];
     }
   }