Changeset 40323 for trunk

Timestamp:

Jan 25, 2018, 6:22:08 AM (8 years ago)

Author:

eugene

Message:

add option to set eta; update nnet test

Location:

trunk/Ohana/src/opihi/cmd.data

Files:

• nnet_train.c (modified) (1 diff)
• test/nnet.sh (modified) (2 diffs)

trunk/Ohana/src/opihi/cmd.data/nnet_train.c

@@ -30 +30 @@
   
   float eta = 1.0; // XXX how do I set this? does it need to update?
+  if ((N = get_argument (argc, argv, "-eta"))) {
+    remove_argument (N, &argc, argv);
+    eta = atof (argv[N]);
+    remove_argument (N, &argc, argv);
+  }
 
   if (argc < 4) {

trunk/Ohana/src/opihi/cmd.data/test/nnet.sh

@@ -126 +126 @@
 
 macro test.simple
+  if ($0 != 4)
+    echo "USAGE: test.simple (Nmini) (Nepoch) (eta)"
+    break
+  end
+
+  local Nmini Nepoch eta
+  $Nmini  = $1
+  $Nepoch = $2
+  $eta    = $3
 
   # create a nnet with just 2 inputs and 2 outputs
@@ -153 +162 @@
 
   nnet create tn 2 2
-  nnet train tn x0 x1 y0 y1 -Nmini 50 -Nepoch 50
+  nnet train tn x0 x1 y0 y1 -Nmini $Nmini -Nepoch $Nepoch -eta $eta
 
   nnet apply tn x0 x1 Y0 Y1
-  lim x0 y0; clear; box; plot -c black x0 y0; plot -c red x0 Y0
+
+  set dy0 = y0 - Y0
+  set dy1 = y1 - Y1
+
+  vstat -q dy0
+  $yp = $MAX
+  $ym = $MIN
+
+  vstat -q dy1
+  $yp = max ($MAX , $yp)
+  $ym = min ($MIN , $ym)
+  $yp = 0.02; $ym = -0.02
+
+  style -pt circle -sz 2; lim x0 $ym $yp; clear; box; plot -c blue x0 dy0; plot -c red x0 dy1
 end