Add 'get_function_activation' function

src/cnn/function.c

@@ -4,6 +4,7 @@
 
 #include "include/function.h"
 
+
 float max(float a, float b) {
     return a < b ? b:a;
 }
@@ -100,3 +101,32 @@ void choose_apply_function_vector(int activation, float*** input, int dim) {
         printf("Erreur, fonction d'activation inconnue (choose_apply_function_vector): %d\n", activation);
     }
 }
+
+ptr get_function_activation(int activation) {
+    if (activation == RELU) {
+        return &relu;
+    } else if (activation == -RELU) {
+        return &relu_derivative;
+    } else if (activation == SIGMOID) {
+        return &sigmoid;
+    } else if (activation == -SIGMOID) {
+        return &sigmoid_derivative;
+    } else if (activation == SOFTMAX) {
+        printf("Erreur, impossible de renvoyer la fonction softmax");
+        return NULL;
+    } else if (activation == -SOFTMAX) {
+        printf("Erreur, impossible de renvoyer la dérivée de la fonction softmax");
+        return NULL;
+    } else if (activation == TANH) {
+        return &tanh_;
+    } else if (activation == -TANH) {
+        return &tanh_derivative;
+    } else {
+        printf("Erreur, fonction d'activation inconnue (choose_apply_function_vector): %d\n", activation);
+        return NULL;
+    }
+}
+// to use: 
+// float a = 5; int activation;
+// pm u = get_function_activation;
+// printf("%f", (*u(activation))(a));

src/cnn/include/function.h

@@ -2,13 +2,15 @@
 #define DEF_FUNCTION_H
 
 
-typedef float (*returnFunctionType)(float, float);
 // Les dérivées sont l'opposé
 #define TANH 1
 #define SIGMOID 2
 #define RELU 3
 #define SOFTMAX 4
 
+typedef float (*ptr)(float);
+typedef ptr (*pm)();
+
 /*
 * Fonction max pour les floats
 */
@@ -46,4 +48,6 @@ void choose_apply_function_matrix(int activation, float*** input, int depth, int
 */
 void choose_apply_function_vector(int activation, float*** input, int dim);
 
+ptr get_function_activation(int activation);
+
 #endif