refactor: Activation and Layer struct refactored

1 files changed, 6 insertions, 2 deletions

diff --git a/src/nn.h b/src/nn.h
index 13b5f7c..0c794b5 100644
--- a/src/nn.h
+++ b/src/nn.h
@@ -10,10 +10,14 @@
 #include <unistd.h>
 #include <openblas/cblas.h>
 
+struct Activation {
+    double (*func)(double);
+    double (*dfunc)(double);
+};
+
 typedef struct Layer {
     double *weights, *bias;
-    double (*activation)(double x);
-    double (*activation_derivative)(double x);
+    struct Activation activation;
     size_t neurons, input_nodes;
 } Layer;