1 files changed, 8 insertions, 3 deletions
diff --git a/code/weightvis.js b/code/weightvis.js
index f61ec39..2ef33e1 100644
--- a/code/weightvis.js
+++ b/code/weightvis.js
@@ -30,7 +30,9 @@ var
     linkHighlightColor    = pDeepPurple[5],
     linkHighlightColorAlt = pPink[5],
     nodeColor             = pDeepPurple[4],
-    nodeHighlightColor    = accentColor
+    nodeColorAlt          = pPink[4],
+    nodeHighlightColor    = accentColor,
+    nodeHighlightColorAlt = pPinkA[3]
 ;
 
 /* network construction */
@@ -63,6 +65,7 @@ var Network = (function() {
                 this.nodes.push({
                     id: layer.toString() + "_" + node.toString(),
                     layer: layer,
+                    node: node,
                     fixed: true
                 });
             }
@@ -507,12 +510,14 @@ var NetVisualizer = (function() {
         function activation(x) { return 1 / (1 + Math.exp(-x)); }
         var weights = this.data[this.layout.data.batches-1][this.layout.data.epochs-1];
         var lastLayerOut = inputs;
+        var activations = [inputs];
         var hasBias = false;
         if (this.layout.net[0].isBias) {
             for (bias=0; bias<inputs.length; bias++) {
                 lastLayerOut[bias] += weights[0][0][bias];
             }
             hasBias = true;
+            activations.push(lastLayerOut);
         }
         for (layer=hasBias?1:0; layer<this.layout.net.layers-1; layer++) {
             // manual vector matrix multiplication
@@ -525,9 +530,9 @@ var NetVisualizer = (function() {
                 thisLayerOut[tar] = activation(thisLayerOut[tar]);
             }
             lastLayerOut = thisLayerOut;
-            //console.log("layer out", layer, lastLayerOut);
+            activations.push(lastLayerOut);
         }
-        return lastLayerOut;
+        return activations;
     };
 
     return NetVisualizer;