up

xieyuheng · May 30, 2024 · f97f624 · f97f624
1 parent 03e0cdc
commit f97f624
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src/system-a/models/iris/irisModel.ts b/src/system-a/models/iris/irisModel.ts
@@ -10,7 +10,12 @@ import { samplingObjective } from "../../tensor/samplingObjective.js"
 import { model } from "../model.js"
 import { irisTrainXs, irisTrainYs } from "./irisDataset.js"
 
-export const irisNetwork = blockStack([denseBlock(4, 6), denseBlock(6, 3)])
+export const irisNetwork = blockStack([
+  denseBlock(4, 6),
+  denseBlock(6, 4),
+  denseBlock(4, 3),
+  denseBlock(3, 3),
+])
 
 export function irisTrainParameters(): Array<Tensor> {
   const objective = samplingObjective(
@@ -29,7 +34,7 @@ export function irisTrainParameters(): Array<Tensor> {
   // })
 
   const gradientDescentFn = gradientDescentAdam({
-    learningRate: 0.002,
+    learningRate: 0.006,
     decayRate: 0.9,
     relayFactor: 0.85,
   })

diff --git a/src/system-a/tensor/zeroTensor.test.ts b/src/system-a/tensor/zeroTensor.test.ts
@@ -2,7 +2,7 @@ import assert from "node:assert"
 import { test } from "node:test"
 import { zeroTensor } from "./zeroTensor.js"
 
-test("tensorZeros", () => {
+test("zeroTensor", () => {
   assert.deepStrictEqual(zeroTensor([2, 3]), [
     [0, 0, 0],
     [0, 0, 0],