From 4a5b6b12040a87dd8319a735cc48f31a2d140e3a Mon Sep 17 00:00:00 2001 From: jvech Date: Thu, 18 Jul 2024 21:42:38 -0500 Subject: add: linear 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 8718a66..f3e6750 100644 --- a/src/main.c +++ b/src/main.c @@ -170,6 +170,7 @@ Layer * load_network(struct Configs cfg) extern struct Activation NN_SOFTPLUS; extern struct Activation NN_SIGMOID; extern struct Activation NN_LEAKY_RELU; + extern struct Activation NN_LINEAR; Layer *network = ecalloc(cfg.network_size, sizeof(Layer)); @@ -178,6 +179,7 @@ Layer * load_network(struct Configs cfg) else if (!strcmp("sigmoid", cfg.activations[i])) network[i].activation = NN_SIGMOID; 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 die("load_network() Error: Unknown '%s' activation", cfg.activations[i]); network[i].neurons = cfg.neurons[i]; -- cgit v1.2.3-70-g09d2