From 3043dc60f1b26d0047dc93a3a1aa758009cbb050 Mon Sep 17 00:00:00 2001 From: Julius Naeumann Date: Sat, 16 Mar 2024 10:08:13 +0100 Subject: [PATCH] tweak --- src/demo/Buddy.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/demo/Buddy.ts b/src/demo/Buddy.ts index 837c13a..837c486 100644 --- 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]}))