Changeset 40376 for trunk/Ohana/src/opihi/cmd.data/nnet_commands.c

Timestamp:

Mar 26, 2018, 12:54:01 PM (8 years ago)

Author:

eugene

Message:

add jdtolst function; add nnet write function

File:

• trunk/Ohana/src/opihi/cmd.data/nnet_commands.c (modified) (1 diff)

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

@@ -258 +258 @@
 }
 
+int nnet_write (int argc, char **argv) {
+
+  if (argc < 3) {
+    gprint (GP_ERR, "USAGE: nnet write (nnet) (filename) : save nnet weights and biases to a file\n");
+    return FALSE;
+  }
+
+  Nnet *nnet = FindNnet (argv[1]);
+  if (nnet == NULL) {
+    gprint (GP_ERR, "nnet %s not found, create it first\n", argv[1]);
+    return FALSE;
+  }
+
+  // open the file and read the number of layers
+  FILE *f = fopen (argv[2], "w");
+  if (f == NULL) {
+    gprint (GP_ERR, "USAGE: nnet write (nnet) (filename) : save nnet weights and biases to a file\n");
+    gprint (GP_ERR, "file %s could not be opened\n", argv[2]);
+    return FALSE;
+  }
+
+  // write the number of layers
+  // NLAYER 3
+  fprintf (f, "NLAYER %d\n", nnet->Nlayer);
+
+  // write the number of nodes for each layer
+  // LAYERS 2 4 2
+  fprintf (f, "LAYERS");
+  for (int i = 0; i < nnet->Nlayer; i++) {
+    fprintf (f, " %d", nnet->Nnodes[i]);
+  }
+  fprintf (f, "\n");
+
+  // read the weights and biases from the data file
+  for (int L = 1; L < nnet[0].Nlayer; L++) {
+    
+    fprintf (f, "LAYER %d NX %d NY %d\n", L - 1, nnet->Nnodes[L-1], nnet->Nnodes[L]);
+
+    for (int j = 0; j < nnet->Nnodes[L]; j++) {
+      for (int i = 0; i < nnet->Nnodes[L-1]; i++) {
+        int k = i + j*nnet[0].Nnodes[L-1];
+        fprintf (f, "%f ", nnet[0].weight[L][k]);
+      }
+      fprintf (f, "%f\n", nnet[0].biases[L][j]);
+    }
+  }
+  fclose (f);
+
+  return TRUE;
+}
+
 int nnet_get (int argc, char **argv) {