up

xieyuheng · Jun 6, 2024 · f87e8fb · f87e8fb
1 parent b4d0d7c
commit f87e8fb
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/src/system-a/models/iris/irisModel.ts b/src/system-a/models/iris/irisModel.ts
@@ -34,9 +34,9 @@ export function irisTrainParameters(): Array<Tensor> {
   const initParameters = denseBlockInitParameters(irisNetwork.shapes)
 
   gradientDescentNaked
-  const gradientDescentFn = gradientDescentNaked({
-    learningRate: 0.0002,
-  })
+  // const gradientDescentFn = gradientDescentNaked({
+  //   learningRate: 0.0002,
+  // })
 
   gradientDescentRms
   // const gradientDescentFn = gradientDescentRms({
@@ -45,11 +45,11 @@ export function irisTrainParameters(): Array<Tensor> {
   // })
 
   gradientDescentAdam
-  // const gradientDescentFn = gradientDescentAdam({
-  //   learningRate: 0.001,
-  //   decayRate: 0.9,
-  //   relayFactor: 0.85,
-  // })
+  const gradientDescentFn = gradientDescentAdam({
+    learningRate: 0.0001,
+    decayRate: 0.9,
+    relayFactor: 0.85,
+  })
 
   return gradientDescentFn(objective, initParameters, {
     revs: 2000,