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

Timestamp:

Jan 27, 2018, 1:09:32 PM (8 years ago)

Author:

eugene

Message:

updated nnet to include better init, L2 regularization, cross-entropy cost function

File:

• trunk/Ohana/src/opihi/cmd.data/nnet_commands.c (modified) (2 diffs)

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

@@ -36 +36 @@
 int nnet_create (int argc, char **argv) {
 
+  int N;
+
+  int LargeWeightInit = FALSE;
+  if ((N = get_argument (argc, argv, "-large-weight-initializer"))) {
+    remove_argument (N, &argc, argv);
+    LargeWeightInit = TRUE;    
+  }
+
   if (argc < 4) {
     gprint (GP_ERR, "USAGE: nnet create (nnet) (Ninput) [Nnodes] [Nnodes] ... (Noutput)\n");
@@ -52 +60 @@
 
   // allocate the vector and matrix data arrays (and init)
-  CreateNnetData (nnet);
+  CreateNnetData (nnet, LargeWeightInit);
 
   return TRUE;