Add layer method

xEcho1337 · Oct 22, 2024 · b420cfb · b420cfb
1 parent 243695c
commit b420cfb
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/src/main/java/net/echo/brain4j/model/Model.java b/src/main/java/net/echo/brain4j/model/Model.java
@@ -80,4 +80,11 @@ public interface Model {
      * @param path path to save model
      */
     void save(String path);
+
+    /**
+     * Adds a layer to the network.
+     *
+     * @param layer the layer to add
+     */
+    void add(Layer layer);
 }
diff --git a/src/main/java/net/echo/brain4j/model/impl/FeedForwardModel.java b/src/main/java/net/echo/brain4j/model/impl/FeedForwardModel.java
@@ -30,6 +30,7 @@
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
+import java.util.concurrent.atomic.AtomicReference;
 
 public class FeedForwardModel implements Model {
 
@@ -110,7 +111,8 @@ public double[] predict(double ... input) {
         Layer inputLayer = layers.get(0);
 
         if (input.length != inputLayer.getNeurons().size()) {
-            throw new IllegalArgumentException("Input size does not match model's input dimension!");
+            throw new IllegalArgumentException("Input size does not match model's input dimension! " +
+                    input.length + " != " + inputLayer.getNeurons().size());
         }
 
         for (Layer layer : layers) {
@@ -278,4 +280,9 @@ public void save(String path) {
             throw new RuntimeException(e);
         }
     }
+
+    @Override
+    public void add(Layer layer) {
+        layers.add(layer);
+    }
 }