mirror of
https://github.com/augustin64/projet-tipe
synced 2025-01-23 23:26:25 +01:00
augustin64
c0381efa71
Ajout de test.cu:
Suppression de redondances dans le code
Recoloration des tests:
Ajout de couleurs pour clarifier le bon fonctionnement ou non du programme
83 lines
2.3 KiB
C
83 lines
2.3 KiB
C
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <math.h>
|
|
#include <stdint.h>
|
|
#include <inttypes.h>
|
|
|
|
#include "../src/mnist/include/neuron_io.h"
|
|
#include "../src/mnist/include/neural_network.h"
|
|
#include "../src/include/colors.h"
|
|
|
|
|
|
Neuron* creer_neuron(int nb_sortants) {
|
|
Neuron* neuron = (Neuron*)malloc(sizeof(Neuron));
|
|
neuron->weights = (float*)malloc(sizeof(float)*nb_sortants);
|
|
neuron->back_weights = (float*)malloc(sizeof(float)*nb_sortants);
|
|
neuron->last_back_weights = (float*)malloc(sizeof(float)*nb_sortants);
|
|
|
|
for (int i=0; i < nb_sortants; i++) {
|
|
neuron->weights[i] = 0.5;
|
|
neuron->back_weights[i] = 0.0;
|
|
neuron->last_back_weights[i] = 0.0;
|
|
}
|
|
neuron->z = 0.0;
|
|
neuron->bias = 0.0;
|
|
neuron->back_bias = 0.0;
|
|
neuron->last_back_bias = 0.0;
|
|
|
|
return neuron;
|
|
}
|
|
|
|
|
|
Layer* creer_layer(int nb_neurons, int nb_sortants) {
|
|
Layer* layer = (Layer*)malloc(sizeof(Layer));
|
|
Neuron** tab = (Neuron**)malloc(sizeof(Neuron*)*nb_neurons);
|
|
|
|
layer->nb_neurons = nb_neurons;
|
|
layer->neurons = tab;
|
|
|
|
for (int i=0; i < nb_neurons; i++) {
|
|
tab[i] = creer_neuron(nb_sortants);
|
|
}
|
|
return layer;
|
|
};
|
|
|
|
|
|
Network* create_network(int nb_layers, int nb_max_neurons, int nb_min_neurons) {
|
|
Network* network = (Network*)malloc(sizeof(Network));
|
|
network->layers = (Layer**)malloc(sizeof(Layer*)*nb_layers);
|
|
int nb_neurons[nb_layers+1];
|
|
|
|
network->nb_layers = nb_layers;
|
|
|
|
for (int i=0; i < nb_layers; i++) {
|
|
nb_neurons[i] = i*(nb_min_neurons-nb_max_neurons)/(nb_layers-1) + nb_max_neurons;
|
|
}
|
|
nb_neurons[nb_layers] = 0;
|
|
|
|
for (int i=0; i < nb_layers; i++) {
|
|
network->layers[i] = creer_layer(nb_neurons[i], nb_neurons[i+1]);
|
|
}
|
|
return network;
|
|
}
|
|
|
|
int main() {
|
|
printf("Création du réseau\n");
|
|
Network* network = create_network(5, 300, 10);
|
|
printf(GREEN "OK\n" RESET);
|
|
|
|
printf("Écriture du réseau\n");
|
|
write_network((char*)".test-cache/neuron_io.bin", network);
|
|
printf(GREEN "OK\n" RESET);
|
|
|
|
printf("Vérification de l'accès en lecture\n");
|
|
Network* network2 = read_network((char*)".test-cache/neuron_io.bin");
|
|
printf(GREEN "OK\n" RESET);
|
|
|
|
printf("Suppression des réseaux\n");
|
|
deletion_of_network(network);
|
|
deletion_of_network(network2);
|
|
printf(GREEN "OK\n" RESET);
|
|
|
|
return 0;
|
|
} |