]> source.dussan.org Git - rspamd.git/commitdiff
[Minor] Set symmetric activation functions for ANN
authorVsevolod Stakhov <vsevolod@highsecure.ru>
Mon, 12 Sep 2016 12:23:25 +0000 (13:23 +0100)
committerVsevolod Stakhov <vsevolod@highsecure.ru>
Mon, 12 Sep 2016 12:23:25 +0000 (13:23 +0100)
src/lua/lua_fann.c

index 0f20983e0568da248c8cd21690b87dbe5605b00c..dd100256e48c90b3660adb40ce5e96a985b768e2 100644 (file)
@@ -113,7 +113,10 @@ lua_fann_create (lua_State *L)
                        layers[i] = luaL_checknumber (L, i + 2);
                }
 
-               f = fann_create_standard_array (nlayers, layers);
+               f = fann_create_shortcut_array (nlayers, layers);
+               fann_set_activation_function_hidden (f, FANN_SIGMOID_SYMMETRIC);
+               fann_set_activation_function_output (f, FANN_SIGMOID_SYMMETRIC);
+               fann_randomize_weights (f, -0.1, 0.1);
 
                if (f != NULL) {
                        pfann = lua_newuserdata (L, sizeof (gpointer));