augustin64/tipe
1
1
Fork 0
mirror of https://github.com/augustin64/projet-tipe synced 2025-01-24 07:36:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update neuron_io

Browse Source
This commit is contained in:
augustin64 2022-05-21 15:11:36 +02:00
parent 45c39d8880
commit 986707af2b
2 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/mnist/include/neuron_io.h
View File

@ -11,8 +11,15 @@
Neuron* read_neuron(uint32_t nb_weights, FILE *ptr);
Neuron** read_neurons(uint32_t nb_neurons, uint32_t nb_weights, FILE *ptr);
Network* read_network(char* filename);
void ecrire_neuron(Neuron* neuron, int weights, FILE *ptr);
void write_neuron(Neuron* neuron, int weights, FILE *ptr);
void write_network(char* filename, Network* network);
Neuron* read_delta_neuron(uint32_t nb_weights, FILE *ptr);
Neuron** read_delta_neurons(uint32_t nb_neurons, uint32_t nb_weights, FILE *ptr);
Network* read_delta_network(char* filename);
void write_delta_neuron(Neuron* neuron, int weights, FILE *ptr);
void write_delta_network(char* filename, Network* network);
#endif

10
src/mnist/neuron_io.c
View File

@ -27,11 +27,9 @@ Neuron* read_neuron(uint32_t nb_weights, FILE *ptr) {
neuron->last_back_bias = 0.0;
neuron->back_bias = 0.0;
float* weights = (float*)malloc(sizeof(float)*nb_weights);
neuron->last_back_weights = (float*)malloc(sizeof(float)*nb_weights);
neuron->back_weights = (float*)malloc(sizeof(float)*nb_weights);
neuron->weights = weights;
neuron->weights = (float*)malloc(sizeof(float)*nb_weights);
for (int i=0; i < (int)nb_weights; i++) {
fread(&tmp, sizeof(float), 1, ptr);
@ -40,7 +38,6 @@ Neuron* read_neuron(uint32_t nb_weights, FILE *ptr) {
neuron->last_back_weights[i] = 0.0;
}
free(weights);
return neuron;
}
@ -160,11 +157,9 @@ Neuron* read_delta_neuron(uint32_t nb_weights, FILE *ptr) {
neuron->last_back_bias = 0.0;
neuron->back_bias = back_bias;
float* back_weights = (float*)malloc(sizeof(float)*nb_weights);
neuron->last_back_weights = (float*)malloc(sizeof(float)*nb_weights);
neuron->back_weights = (float*)malloc(sizeof(float)*nb_weights);
neuron->weights = back_weights;
neuron->weights = (float*)malloc(sizeof(float)*nb_weights);
for (int i=0; i < (int)nb_weights; i++) {
fread(&tmp, sizeof(float), 1, ptr);
@ -172,7 +167,6 @@ Neuron* read_delta_neuron(uint32_t nb_weights, FILE *ptr) {
neuron->back_weights[i] = tmp;
neuron->last_back_weights[i] = 0.0;
}
free(back_weights);
return neuron;
}