From 13d2f9ee07a632ab09ed4eacabc57ce406a2dff0 Mon Sep 17 00:00:00 2001 From: jvech Date: Fri, 19 Jul 2024 09:13:18 -0500 Subject: add: tanh activation added --- src/main.c | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/main.c') diff --git a/src/main.c b/src/main.c index f3e6750..dab8bd9 100644 --- a/src/main.c +++ b/src/main.c @@ -171,6 +171,7 @@ Layer * load_network(struct Configs cfg) extern struct Activation NN_SIGMOID; extern struct Activation NN_LEAKY_RELU; extern struct Activation NN_LINEAR; + extern struct Activation NN_TANH; Layer *network = ecalloc(cfg.network_size, sizeof(Layer)); @@ -180,6 +181,7 @@ Layer * load_network(struct Configs cfg) else if (!strcmp("softplus", cfg.activations[i])) network[i].activation = NN_SOFTPLUS; else if (!strcmp("leaky_relu", cfg.activations[i])) network[i].activation = NN_LEAKY_RELU; else if (!strcmp("linear", cfg.activations[i])) network[i].activation = NN_LINEAR; + else if (!strcmp("tanh", cfg.activations[i])) network[i].activation = NN_TANH; else die("load_network() Error: Unknown '%s' activation", cfg.activations[i]); network[i].neurons = cfg.neurons[i]; -- cgit v1.2.3-70-g09d2