QOL changes

xEcho1337 · Dec 3, 2024 · 64e52e1 · 64e52e1
1 parent 8820870
commit 64e52e1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/src/main/java/net/echo/brain4j/utils/Vector.java b/src/main/java/net/echo/brain4j/utils/Vector.java
@@ -19,6 +19,10 @@ public static Vector of(double... data) {
         return new Vector(data);
     }
 
+    public static Vector random(int size) {
+        return new Vector(size).fill(Math::random);
+    }
+
     public void set(int index, double value) {
         data[index] = value;
     }
@@ -101,16 +105,20 @@ public Vector convoluted(Vector other) {
         return new Vector(result);
     }
 
-    public void add(Vector other) {
+    public Vector add(Vector other) {
         for (int i = 0; i < data.length; i++) {
             data[i] += other.data[i];
         }
+
+        return this;
     }
 
-    public void scale(double value) {
+    public Vector scale(double value) {
         for (int i = 0; i < data.length; i++) {
             data[i] *= value;
         }
+
+        return this;
     }
 
     public double[] toArray() {

diff --git a/src/test/java/antiswear/ToxicCommentClassification.java b/src/test/java/antiswear/ToxicCommentClassification.java
@@ -71,11 +71,7 @@ private static Map<String, Vector> loadVocab() {
             for (String token : content) {
                 if (token.startsWith("[") || token.startsWith("##")) continue;
 
-                Vector embedding = new Vector(EMBEDDING_DIM);
-
-                embedding.fill(Math::random).scale(10);
-
-                vectors.put(token, embedding);
+                vectors.put(token, Vector.random(EMBEDDING_DIM).scale(10));
             }
         } catch (Exception e) {
             e.printStackTrace(System.err);