1 files changed, 8 insertions, 1 deletions
diff --git a/code/weightvis.js b/code/weightvis.js
index 428d3ce..b964367 100644
--- a/code/weightvis.js
+++ b/code/weightvis.js
@@ -517,7 +517,14 @@ var NetVisualizer = (function() {
             min = d3.min(flatActivations, Math.abs);
         var node = this.svg.selectAll(".node");
         node.each(function(d) { d.altLayout = activations[d.layer][d.node]>0; })
-            .attr("r",  function(d) { return 2 * d.r * (Math.abs(activations[d.layer][d.node]) - min) / (max-min); })
+            .attr("r", function(d) { return 2 * d.r * (Math.abs(activations[d.layer][d.node]) - min) / (max-min); })
+            .style("fill", fNodeColor);
+    };
+
+    NetVisualizer.prototype.clearNodeHighlights = function() {
+        this.svg.selectAll(".node")
+            .each(function(d) { d.altLayout = false; })
+            .attr("r", function(d) { return d.r; })
             .style("fill", fNodeColor);
     };