tweak

JuliusNmn · Mar 16, 2024 · 3043dc6 · 3043dc6
1 parent 989751c
commit 3043dc6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/demo/Buddy.ts b/src/demo/Buddy.ts
@@ -12,7 +12,7 @@ export class Brain {
     const decisionOutputDimension = this.io.muscles.length;
 
     this.model = tf.sequential();
-    const hiddenLayer = 15;
+    const hiddenLayer = 30;
     const A1 = tf.randomNormal([stateInputDimension, hiddenLayer], 0, 0.5)
     const B1 = tf.randomUniform([hiddenLayer], -0.5, 0.5)
     this.model.add(tf.layers.dense({units: hiddenLayer, inputShape: [stateInputDimension], weights: [A1, B1]}))