From f97f624c2bc41702975bf53999a96b7a1f69e39f Mon Sep 17 00:00:00 2001
From: Xie Yuheng <xyheme@gmail.com>
Date: Fri, 31 May 2024 04:29:50 +0800
Subject: [PATCH] up

---
 src/system-a/models/iris/irisModel.ts  | 9 +++++++--
 src/system-a/tensor/zeroTensor.test.ts | 2 +-
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/system-a/models/iris/irisModel.ts b/src/system-a/models/iris/irisModel.ts
index de2ceff..d5ad1e7 100644
--- 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
index 70e7508..c2e0749 100644
--- 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],