From e280d3e9da6424dd6ddbec4f8de89812357acfbc Mon Sep 17 00:00:00 2001
From: Julien Chemillier <chemillier.julien@gmail.com>
Date: Fri, 9 Sep 2022 17:39:07 +0200
Subject: [PATCH] Update mnist_cnn: improve code readability

---
 src/mnist_cnn/cnn.c            | 52 ++++++++++++-------
 src/mnist_cnn/cnn.h            |  4 +-
 src/mnist_cnn/creation.c       | 93 ++++++++++++++++------------------
 src/mnist_cnn/free.c           | 23 ++++-----
 src/mnist_cnn/function.c       | 28 +++++-----
 src/mnist_cnn/initialisation.c | 24 ++++-----
 src/mnist_cnn/make.c           | 44 ++++++++--------
 7 files changed, 136 insertions(+), 132 deletions(-)

diff --git a/src/mnist_cnn/cnn.c b/src/mnist_cnn/cnn.c
index ee5ca40..a81879f 100644
--- a/src/mnist_cnn/cnn.c
+++ b/src/mnist_cnn/cnn.c
@@ -2,21 +2,24 @@
 #include <stdio.h>
 #include <math.h>
 #include <float.h>
-#include "function.h"
-#include "make.h"
+#include "initialisation.c"
+#include "function.c"
+#include "creation.c"
+#include "make.c"
+
 #include "cnn.h"
 
 // Augmente les dimensions de l'image d'entrée
-#define PADING_INPUT 2
+#define PADDING_INPUT 2
 
 int will_be_drop(int dropout_prob) {
-    return (rand() % 100)<dropout_prob;
+    return (rand() % 100) < dropout_prob;
 }
 
-void write_image_in_newtork_32(int** image, int height, int width, float** input) {
-    for (int i=0; i < height+2*PADING_INPUT; i++) {
-        for (int j=PADING_INPUT; j < width+2*PADING_INPUT; j++) {
-            if (i<PADING_INPUT || i>height+PADING_INPUT || j<PADING_INPUT || j>width+PADING_INPUT) {
+void write_image_in_network_32(int** image, int height, int width, float** input) {
+    for (int i=0; i < height+2*PADDING_INPUT; i++) {
+        for (int j=PADDING_INPUT; j < width+2*PADDING_INPUT; j++) {
+            if (i < PADDING_INPUT || i > height+PADDING_INPUT || j < PADDING_INPUT || j > width+PADDING_INPUT) {
                 input[i][j] = 0.;
             }
             else {
@@ -27,16 +30,21 @@ void write_image_in_newtork_32(int** image, int height, int width, float** input
 }
 
 void forward_propagation(Network* network) {
+    int output_dim, output_depth;
+    float*** output;
     for (int i=0; i < network->size-1; i++) {
-        if (network->kernel[i].nn==NULL && network->kernel[i].cnn!=NULL) {
-            make_convolution(network->input[i], network->kernel[i].cnn, network->input[i+1], network->dim[i+1][0]);
-            choose_apply_function_input(network->kernel[i].activation, network->input[i+1], network->dim[i+1][1], network->dim[i+1][0], network->dim[i+1][0]);
+        if (network->kernel[i].nn==NULL && network->kernel[i].cnn!=NULL) { //CNN
+            output = network->input[i+1];
+            output_dim = network->dim[i+1][0];
+            output_depth = network->dim[i+1][1];
+            make_convolution(network->input[i], network->kernel[i].cnn, output, output_dim);
+            choose_apply_function_input(network->kernel[i].activation, output, output_depth, output_dim, output_dim);
         }
-        else if (network->kernel[i].nn!=NULL && network->kernel[i].cnn==NULL) {
+        else if (network->kernel[i].nn!=NULL && network->kernel[i].cnn==NULL) { //NN
             make_fully_connected(network->input[i][0][0], network->kernel[i].nn, network->input[i+1][0][0], network->dim[i][0], network->dim[i+1][0]);
             choose_apply_function_input(network->kernel[i].activation, network->input[i+1], 1, 1, network->dim[i+1][0]);
         }
-        else {
+        else { //Pooling
             if (network->size-2==i) {
                 printf("Le réseau ne peut pas finir par une pooling layer");
                 return;
@@ -61,13 +69,12 @@ void backward_propagation(Network* network, float wanted_number) {
     float* wanted_output = generate_wanted_output(wanted_number);
     int n = network->size-1;
     float loss = compute_cross_entropy_loss(network->input[n][0][0], wanted_output, network->dim[n][0]);
-    int i, j;
-    for (i=n; i>=0; i--) {
+    for (int i=n; i >= 0; i--) {
         if (i==n) {
             if (network->kernel[i].activation == SOFTMAX) {
                 int l2 = network->dim[i][0]; // Taille de la dernière couche
                 int l1 = network->dim[i-1][0];
-                for (j=0; j<l2; j++) {
+                for (int j=0; j < l2; j++) {
 
                 }
             }
@@ -93,7 +100,7 @@ void backward_propagation(Network* network, float wanted_number) {
 
 float compute_cross_entropy_loss(float* output, float* wanted_output, int len) {
     float loss=0.;
-    for (int i=0; i<len ; i++) {
+    for (int i=0; i < len ; i++) {
         if (wanted_output[i]==1) {
             if (output[i]==0.) {
                 loss -= log(FLT_EPSILON);
@@ -107,8 +114,8 @@ float compute_cross_entropy_loss(float* output, float* wanted_output, int len) {
 }
                 
 float* generate_wanted_output(float wanted_number) {
-    float* wanted_output = malloc(sizeof(float)*10);
-    for (int i=0; i<10; i++) {
+    float* wanted_output = (float*)malloc(sizeof(float)*10);
+    for (int i=0; i < 10; i++) {
         if (i==wanted_number) {
             wanted_output[i]=1;
         }
@@ -118,3 +125,10 @@ float* generate_wanted_output(float wanted_number) {
     }
     return wanted_output;
 }
+
+int main() {
+    Network* network;
+    network = create_network_lenet5(0, TANH, GLOROT_NORMAL);
+    forward_propagation(network);
+    return 0;
+}
\ No newline at end of file
diff --git a/src/mnist_cnn/cnn.h b/src/mnist_cnn/cnn.h
index d4525ab..da90422 100644
--- a/src/mnist_cnn/cnn.h
+++ b/src/mnist_cnn/cnn.h
@@ -10,9 +10,9 @@
 int will_be_drop(int dropout_prob);
 
 /*
-* Ecrit une image 28*28 au centre d'un tableau 32*32 et met à 0 le reste
+* Écrit une image 28*28 au centre d'un tableau 32*32 et met à 0 le reste
 */
-void write_image_in_newtork_32(int** image, int height, int width, float** input);
+void write_image_in_network_32(int** image, int height, int width, float** input);
 
 /*
 * Propage en avant le cnn
diff --git a/src/mnist_cnn/creation.c b/src/mnist_cnn/creation.c
index 53a3b76..bf7938d 100644
--- a/src/mnist_cnn/creation.c
+++ b/src/mnist_cnn/creation.c
@@ -5,24 +5,23 @@
 #include "initialisation.h"
 
 Network* create_network(int max_size, int dropout, int initialisation, int input_dim, int input_depth) {
-    if (dropout<0 || dropout>100) {
+    if (dropout < 0 || dropout > 100) {
         printf("Erreur, la probabilité de dropout n'est pas respecté, elle doit être comprise entre 0 et 100\n");
     }
-    Network* network = malloc(sizeof(Network));
+    Network* network = (Network*)malloc(sizeof(Network));
     network->max_size = max_size;
     network->dropout = dropout;
     network->initialisation = initialisation;
     network->size = 1;
-    network->input = malloc(sizeof(float***)*max_size);
-    network->kernel = malloc(sizeof(Kernel)*(max_size-1));
-    create_a_cube_input_layer(network, 0, input_depth, input_dim);
-    int i, j;
-    network->dim = malloc(sizeof(int*)*max_size);
-    for (i=0; i<max_size; i++) {
-        network->dim[i] = malloc(sizeof(int)*2);
+    network->input = (float****)malloc(sizeof(float***)*max_size);
+    network->kernel = (Kernel*)malloc(sizeof(Kernel)*(max_size-1));
+    network->dim = (int**)malloc(sizeof(int*)*max_size);
+    for (int i=0; i < max_size; i++) {
+        network->dim[i] = (int*)malloc(sizeof(int)*2);
     }
     network->dim[0][0] = input_dim;
     network->dim[0][1] = input_depth;
+    create_a_cube_input_layer(network, 0, input_depth, input_dim);
     return network;
 }
 
@@ -40,12 +39,11 @@ Network* create_network_lenet5(int dropout, int activation, int initialisation)
 }
 
 void create_a_cube_input_layer(Network* network, int pos, int depth, int dim) {
-    int i, j;
-    network->input[pos] = malloc(sizeof(float**)*depth);
-    for (i=0; i<depth; i++) {
-        network->input[pos][i] = malloc(sizeof(float*)*dim);
-        for (j=0; j<dim; j++) {
-            network->input[pos][i][j] = malloc(sizeof(float)*dim);
+    network->input[pos] = (float***)malloc(sizeof(float**)*depth);
+    for (int i=0; i < depth; i++) {
+        network->input[pos][i] = (float**)malloc(sizeof(float*)*dim);
+        for (int j=0; j < dim; j++) {
+            network->input[pos][i][j] = (float*)malloc(sizeof(float)*dim);
         }
     }
     network->dim[pos][0] = dim;
@@ -53,10 +51,9 @@ void create_a_cube_input_layer(Network* network, int pos, int depth, int dim) {
 }
 
 void create_a_line_input_layer(Network* network, int pos, int dim) {
-    int i;
-    network->input[pos] = malloc(sizeof(float**));
-    network->input[pos][0] = malloc(sizeof(float*));
-    network->input[pos][0][0] = malloc(sizeof(float)*dim);
+    network->input[pos] = (float***)malloc(sizeof(float**));
+    network->input[pos][0] = (float**)malloc(sizeof(float*));
+    network->input[pos][0][0] = (float*)malloc(sizeof(float)*dim);
 }
 
 void add_average_pooling(Network* network, int kernel_size, int activation) {
@@ -87,7 +84,7 @@ void add_average_pooling_flatten(Network* network, int kernel_size, int activati
 }
 
 void add_convolution(Network* network, int nb_filter, int kernel_size, int activation) {
-    int n = network->size, i, j, k;
+    int n = network->size;
     if (network->max_size == n) {
         printf("Impossible de rajouter une couche de convolution, le réseau est déjà plein\n");
         return;
@@ -95,33 +92,33 @@ void add_convolution(Network* network, int nb_filter, int kernel_size, int activ
     int r = network->dim[n-1][1];
     int c = nb_filter;
     network->kernel[n].nn = NULL;
-    network->kernel[n].cnn = malloc(sizeof(Kernel_cnn));
+    network->kernel[n].cnn = (Kernel_cnn*)malloc(sizeof(Kernel_cnn));
     network->kernel[n].activation = activation;
     network->kernel[n].cnn->k_size = kernel_size;
     network->kernel[n].cnn->rows = r;
     network->kernel[n].cnn->columns = c;
-    network->kernel[n].cnn->w = malloc(sizeof(float***)*r);
-    network->kernel[n].cnn->d_w = malloc(sizeof(float***)*r);
-    for (i=0; i<r; i++) {
-        network->kernel[n].cnn->w[i] = malloc(sizeof(float**)*c);
-        network->kernel[n].cnn->d_w[i] = malloc(sizeof(float**)*c);
-        for (j=0; j<c; j++) {
-            network->kernel[n].cnn->w[i][j] = malloc(sizeof(float*)*kernel_size);
-            network->kernel[n].cnn->d_w[i][j] = malloc(sizeof(float*)*kernel_size);
-            for (k=0; k<kernel_size; k++) {
-                network->kernel[n].cnn->w[i][j][k] = malloc(sizeof(float)*kernel_size);
-                network->kernel[n].cnn->d_w[i][j][k] = malloc(sizeof(float)*kernel_size);
+    network->kernel[n].cnn->w = (float****)malloc(sizeof(float***)*r);
+    network->kernel[n].cnn->d_w = (float****)malloc(sizeof(float***)*r);
+    for (int i=0; i < r; i++) {
+        network->kernel[n].cnn->w[i] = (float***)malloc(sizeof(float**)*c);
+        network->kernel[n].cnn->d_w[i] = (float***)malloc(sizeof(float**)*c);
+        for (int j=0; j < c; j++) {
+            network->kernel[n].cnn->w[i][j] = (float**)malloc(sizeof(float*)*kernel_size);
+            network->kernel[n].cnn->d_w[i][j] = (float**)malloc(sizeof(float*)*kernel_size);
+            for (int k=0; k < kernel_size; k++) {
+                network->kernel[n].cnn->w[i][j][k] = (float*)malloc(sizeof(float)*kernel_size);
+                network->kernel[n].cnn->d_w[i][j][k] = (float*)malloc(sizeof(float)*kernel_size);
             }
         }
     }
-    network->kernel[n].cnn->bias = malloc(sizeof(float**)*c);
-    network->kernel[n].cnn->d_bias = malloc(sizeof(float**)*c);
-    for (i=0; i<c; i++) {
-        network->kernel[n].cnn->bias[i] = malloc(sizeof(float*)*kernel_size);
-        network->kernel[n].cnn->d_bias[i] = malloc(sizeof(float*)*kernel_size);
-        for (j=0; j<kernel_size; j++) {
-            network->kernel[n].cnn->bias[i][j] = malloc(sizeof(float)*kernel_size);
-            network->kernel[n].cnn->d_bias[i][j] = malloc(sizeof(float)*kernel_size);
+    network->kernel[n].cnn->bias = (float***)malloc(sizeof(float**)*c);
+    network->kernel[n].cnn->d_bias = (float***)malloc(sizeof(float**)*c);
+    for (int i=0; i < c; i++) {
+        network->kernel[n].cnn->bias[i] = (float**)malloc(sizeof(float*)*kernel_size);
+        network->kernel[n].cnn->d_bias[i] = (float**)malloc(sizeof(float*)*kernel_size);
+        for (int j=0; j < kernel_size; j++) {
+            network->kernel[n].cnn->bias[i][j] = (float*)malloc(sizeof(float)*kernel_size);
+            network->kernel[n].cnn->d_bias[i][j] = (float*)malloc(sizeof(float)*kernel_size);
         }
     }
     create_a_cube_input_layer(network, n, c, network->dim[n-1][0] - 2*(kernel_size/2));
@@ -141,17 +138,17 @@ void add_dense(Network* network, int input_units, int output_units, int activati
         return;
     }
     network->kernel[n].cnn = NULL;
-    network->kernel[n].nn = malloc(sizeof(Kernel_nn));
+    network->kernel[n].nn = (Kernel_nn*)malloc(sizeof(Kernel_nn));
     network->kernel[n].activation = activation;
     network->kernel[n].nn->input_units = input_units;
     network->kernel[n].nn->output_units = output_units;
-    network->kernel[n].nn->bias = malloc(sizeof(float)*output_units);
-    network->kernel[n].nn->d_bias = malloc(sizeof(float)*output_units);
-    network->kernel[n].nn->weights = malloc(sizeof(float*)*input_units);
-    network->kernel[n].nn->d_weights = malloc(sizeof(float*)*input_units);
-    for (int i=0; i<input_units; i++) {
-        network->kernel[n].nn->weights[i] = malloc(sizeof(float)*output_units);
-        network->kernel[n].nn->d_weights[i] = malloc(sizeof(float)*output_units);
+    network->kernel[n].nn->bias = (float*)malloc(sizeof(float)*output_units);
+    network->kernel[n].nn->d_bias = (float*)malloc(sizeof(float)*output_units);
+    network->kernel[n].nn->weights = (float**)malloc(sizeof(float*)*input_units);
+    network->kernel[n].nn->d_weights = (float**)malloc(sizeof(float*)*input_units);
+    for (int i=0; i < input_units; i++) {
+        network->kernel[n].nn->weights[i] = (float*)malloc(sizeof(float)*output_units);
+        network->kernel[n].nn->d_weights[i] = (float*)malloc(sizeof(float)*output_units);
     }
     initialisation_1d_matrix(network->initialisation, network->kernel[n].nn->bias, output_units, output_units+input_units);
     initialisation_1d_matrix(ZERO, network->kernel[n].nn->d_bias, output_units, output_units+input_units);
diff --git a/src/mnist_cnn/free.c b/src/mnist_cnn/free.c
index cb6a293..64980a7 100644
--- a/src/mnist_cnn/free.c
+++ b/src/mnist_cnn/free.c
@@ -3,9 +3,8 @@
 #include "free.h"
 
 void free_a_cube_input_layer(Network* network, int pos, int depth, int dim) {
-    int i, j, k;
-    for (i=0; i<depth; i++) {
-        for (j=0; j<dim; j++) {
+    for (int i=0; i < depth; i++) {
+        for (int j=0; j < dim; j++) {
             free(network->input[pos][i][j]);
         }
         free(network->input[pos][i]);
@@ -28,12 +27,12 @@ void free_average_pooling_flatten(Network* network, int pos) {
 }
 
 void free_convolution(Network* network, int pos) {
-    int i, j, k, c = network->kernel[pos].cnn->columns;
+    int c = network->kernel[pos].cnn->columns;
     int k_size = network->kernel[pos].cnn->k_size;
     int r = network->kernel[pos].cnn->rows;
     free_a_cube_input_layer(network, pos, c, network->dim[pos-1][0] - 2*(k_size/2));
-    for (i=0; i<c; i++) {
-        for (j=0; j<k_size; j++) {
+    for (int i=0; i < c; i++) {
+        for (int j=0; j < k_size; j++) {
             free(network->kernel[pos].cnn->bias[i][j]);
             free(network->kernel[pos].cnn->d_bias[i][j]);
         }
@@ -43,9 +42,9 @@ void free_convolution(Network* network, int pos) {
     free(network->kernel[pos].cnn->bias);
     free(network->kernel[pos].cnn->d_bias);
 
-    for (i=0; i<r; i++) {
-        for (j=0; j<c; j++) {
-            for (k=0; k<k_size; k++) {
+    for (int i=0; i < r; i++) {
+        for (int j=0; j < c; j++) {
+            for (int k=0; k < k_size; k++) {
                 free(network->kernel[pos].cnn->w[i][j][k]);
                 free(network->kernel[pos].cnn->d_w[i][j][k]);
             }
@@ -63,8 +62,8 @@ void free_convolution(Network* network, int pos) {
 
 void free_dense(Network* network, int pos) {
     free_a_line_input_layer(network, pos);
-    int i, dim = network->kernel[pos].nn->output_units;
-    for (int i=0; i<dim; i++) {
+    int dim = network->kernel[pos].nn->output_units;
+    for (int i=0; i < dim; i++) {
         free(network->kernel[pos].nn->weights[i]);
         free(network->kernel[pos].nn->d_weights[i]);
     }
@@ -80,7 +79,7 @@ void free_dense(Network* network, int pos) {
 void free_network_creation(Network* network) {
     free_a_cube_input_layer(network, 0, network->dim[0][1], network->dim[0][0]);
 
-    for (int i=0; i<network->max_size; i++) {
+    for (int i=0; i < network->max_size; i++) {
         free(network->dim[i]);
     }
     free(network->dim);
diff --git a/src/mnist_cnn/function.c b/src/mnist_cnn/function.c
index 5f2bb82..360af32 100644
--- a/src/mnist_cnn/function.c
+++ b/src/mnist_cnn/function.c
@@ -4,7 +4,7 @@
 #include "function.h"
 
 float max(float a, float b) {
-    return a<b?b:a;
+    return a < b ? b:a;
 }
 
 float sigmoid(float x) {
@@ -36,27 +36,26 @@ float tanh_derivative(float x) {
 }
 
 void apply_softmax_input(float ***input, int depth, int rows, int columns) {
-    int i, j, k;
     float m = FLT_MIN;
     float sum=0;
-    for (i=0; i<depth; i++) {
-        for (j=0; j<rows; j++) {
-            for (k=0; k<columns; k++) {
+    for (int i=0; i < depth; i++) {
+        for (int j=0; j < rows; j++) {
+            for (int k=0; k < columns; k++) {
                 m = max(m, input[i][j][k]);
             }
         }
     }
-    for (i=0; i<depth; i++) {
-        for (j=0; j<rows; j++) {
-            for (k=0; k<columns; k++) {
+    for (int i=0; i < depth; i++) {
+        for (int j=0; j < rows; j++) {
+            for (int k=0; k < columns; k++) {
                 input[i][j][k] = exp(m-input[i][j][k]);
                 sum += input[i][j][k];
             }
         }
     }
-    for (i=0; i<depth; i++) {
-        for (j=0; j<rows; j++) {
-            for (k=0; k<columns; k++) {
+    for (int i=0; i < depth; i++) {
+        for (int j=0; j < rows; j++) {
+            for (int k=0; k < columns; k++) {
                 input[i][j][k] = input[i][j][k]/sum;
             }
         }
@@ -64,10 +63,9 @@ void apply_softmax_input(float ***input, int depth, int rows, int columns) {
 }
 
 void apply_function_input(float (*f)(float), float*** input, int depth, int rows, int columns) {
-    int i, j ,k;
-    for (i=0; i<depth; i++) {
-        for (j=0; j<rows; j++) {
-            for (k=0; k<columns; k++) {
+    for (int i=0; i < depth; i++) {
+        for (int j=0; j < rows; j++) {
+            for (int k=0; k < columns; k++) {
                 input[i][j][k] = (*f)(input[i][j][k]);
             }
         }
diff --git a/src/mnist_cnn/initialisation.c b/src/mnist_cnn/initialisation.c
index 2c5de09..ef2d24a 100644
--- a/src/mnist_cnn/initialisation.c
+++ b/src/mnist_cnn/initialisation.c
@@ -4,32 +4,29 @@
 
 
 void initialisation_1d_matrix(int initialisation, float* matrix, int rows, int n) { //NOT FINISHED
-    int i;
     float lower_bound = -6/sqrt((double)n);
     float distance = -lower_bound-lower_bound;
-    for (i=0; i<rows; i++) {
+    for (int i=0; i < rows; i++) {
         matrix[i] = lower_bound + RAND_FLT()*distance;
     }
 }
 
 void initialisation_2d_matrix(int initialisation, float** matrix, int rows, int columns, int n) { //NOT FINISHED
-    int i, j;
     float lower_bound = -6/sqrt((double)n);
     float distance = -lower_bound-lower_bound;
-    for (i=0; i<rows; i++) {
-        for (j=0; j<columns; j++) {
+    for (int i=0; i < rows; i++) {
+        for (int j=0; j < columns; j++) {
             matrix[i][j] = lower_bound + RAND_FLT()*distance;
         }
     }
 }
 
 void initialisation_3d_matrix(int initialisation, float*** matrix, int depth, int rows, int columns, int n) { //NOT FINISHED
-    int i, j, k;
     float lower_bound = -6/sqrt((double)n);
     float distance = -lower_bound-lower_bound;
-    for (i=0; i<depth; i++) {
-        for (j=0; j<rows; j++) {
-            for (k=0; k<columns; k++) {
+    for (int i=0; i < depth; i++) {
+        for (int j=0; j < rows; j++) {
+            for (int k=0; k < columns; k++) {
                 matrix[i][j][k] = lower_bound + RAND_FLT()*distance;
             }
         }
@@ -37,13 +34,12 @@ void initialisation_3d_matrix(int initialisation, float*** matrix, int depth, in
 }
 
 void initialisation_4d_matrix(int initialisation, float**** matrix, int rows, int columns, int rows1, int columns1, int n) { //NOT FINISHED
-    int i, j, k, l;
     float lower_bound = -6/sqrt((double)n);
     float distance = -lower_bound-lower_bound;
-    for (i=0; i<rows; i++) {
-        for (j=0; j<columns; j++) {
-            for (k=0; k<rows1; k++) {
-                for (l=0; l<columns1; l++) {
+    for (int i=0; i < rows; i++) {
+        for (int j=0; j < columns; j++) {
+            for (int k=0; k < rows1; k++) {
+                for (int l=0; l < columns1; l++) {
                     matrix[i][j][k][l] = lower_bound + RAND_FLT()*distance;
                 }
             }
diff --git a/src/mnist_cnn/make.c b/src/mnist_cnn/make.c
index c02a815..15be0d1 100644
--- a/src/mnist_cnn/make.c
+++ b/src/mnist_cnn/make.c
@@ -4,14 +4,14 @@
 void make_convolution(float*** input, Kernel_cnn* kernel, float*** output, int output_dim) {
     //NOT FINISHED, MISS CONDITIONS ON THE CONVOLUTION
     float f;
-    int i, j, k, a, b, c, n=kernel->k_size;
-    for (i=0; i<kernel->columns; i++) {
-        for (j=0; j<output_dim; j++) {
-            for (k=0; k<output_dim; k++) {
+    int n = kernel->k_size;
+    for (int i=0; i < kernel->columns; i++) {
+        for (int j=0; j < output_dim; j++) {
+            for (int k=0; k < output_dim; k++) {
                 f = kernel->bias[i][j][k];
-                for (a=0; a<kernel->rows; a++) {
-                    for (b=0; b<n; b++) {
-                        for (c=0; c<n; c++) {
+                for (int a=0; a < kernel->rows; a++) {
+                    for (int b=0; b < n; b++) {
+                        for (int c=0; c < n; c++) {
                             f += kernel->w[a][i][b][c]*input[a][j+a][k+b];
                         }
                     }
@@ -25,13 +25,13 @@ void make_convolution(float*** input, Kernel_cnn* kernel, float*** output, int o
 void make_average_pooling(float*** input, float*** output, int size, int output_depth, int output_dim) {
     //NOT FINISHED, MISS CONDITIONS ON THE POOLING
     float average;
-    int i, j, k, a, b, n=size*size;
-    for (i=0; i<output_depth; i++) {
-        for (j=0; j<output_dim; j++) {
-            for (k=0; k<output_dim; k++) {
+    int n = size*size;
+    for (int i=0; i < output_depth; i++) {
+        for (int j=0; j < output_dim; j++) {
+            for (int k=0; k < output_dim; k++) {
                 average = 0.;
-                for (a=0; a<size; a++) {
-                    for (b=0; b<size; b++) {
+                for (int a=0; a < size; a++) {
+                    for (int b=0; b < size; b++) {
                         average += input[i][2*j +a][2*k +b];
                     }
                 }
@@ -47,14 +47,15 @@ void make_average_pooling_flattened(float*** input, float* output, int size, int
         return;
     }
     float average;
-    int i, j, k, a, b, n=size*size, cpt=0;
+    int n = size*size;
+    int cpt = 0;
     int output_dim = input_dim - 2*(size/2);
-    for (i=0; i<input_depth; i++) {
-        for (j=0; j<output_dim; j++) {
-            for (k=0; k<output_dim; k++) {
+    for (int i=0; i < input_depth; i++) {
+        for (int j=0; j < output_dim; j++) {
+            for (int k=0; k < output_dim; k++) {
                 average = 0.;
-                for (a=0; a<size; a++) {
-                    for (b=0; b<size; b++) {
+                for (int a=0; a < size; a++) {
+                    for (int b=0; b < size; b++) {
                         average += input[i][2*j +a][2*k +b];
                     }
                 }
@@ -66,11 +67,10 @@ void make_average_pooling_flattened(float*** input, float* output, int size, int
 }
 
 void make_fully_connected(float* input, Kernel_nn* kernel, float* output, int size_input, int size_output) {
-    int i, j, k;
     float f;
-    for (i=0; i<size_output; i++) {
+    for (int i=0; i < size_output; i++) {
         f = kernel->bias[i];
-        for (j=0; j<size_input; j++) {
+        for (int j=0; j < size_input; j++) {
             f += kernel->weights[i][j]*input[j];
         }
         output[i] = f;