Update dann.js

matiasvlevi · Sep 4, 2020 · 81028da · 81028da
1 parent 1637aad
commit 81028da
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/dann.js b/dann.js
@@ -135,12 +135,17 @@ class DANNeuralNetwork {
         console.log("  Layers:")
         for (let i = 0; i < this.Layers.length;i++) {
             let str = "Hidden Layer: ";
+            let afunc = "";
             if (i == 0) {
                 str = "Input Layer: ";
+                afunc = "      ";
             } else if (i == this.Layers.length-1) {
                 str = "Output Layer: ";
+                afunc = this.aFunc[i+1].name;
+            } else {
+                afunc = this.aFunc[i+1].name;
             }
-            console.log("    " + this.aFunc[i].name + " " + str + Matrix.toArray(this.Layers[i]).length);
+            console.log("    " + afunc + " " + str + Matrix.toArray(this.Layers[i]).length);
         }
         console.log(" ");
         console.log("  Learning rate:")