Change 'wanted_number' from float to int

This commit is contained in:
julienChemillier 2023-01-18 10:25:46 +01:00
parent cd54fba17d
commit a21481d1cc
2 changed files with 7 additions and 7 deletions

View File

@ -126,8 +126,8 @@ void forward_propagation(Network* network) {
} }
} }
void backward_propagation(Network* network, float wanted_number) { void backward_propagation(Network* network, int wanted_number) {
float* wanted_output = generate_wanted_output(wanted_number); float* wanted_output = generate_wanted_output(wanted_number, 10);
int n = network->size; int n = network->size;
int activation, input_depth, input_width, output_depth, output_width; int activation, input_depth, input_width, output_depth, output_width;
float*** input; float*** input;
@ -215,9 +215,9 @@ float compute_cross_entropy_loss(float* output, float* wanted_output, int len) {
return loss; return loss;
} }
float* generate_wanted_output(float wanted_number) { float* generate_wanted_output(int wanted_number, int size_output) {
float* wanted_output = (float*)malloc(sizeof(float)*10); float* wanted_output = (float*)malloc(sizeof(float)*size_output);
for (int i=0; i < 10; i++) { for (int i=0; i < size_output; i++) {
if (i==wanted_number) { if (i==wanted_number) {
wanted_output[i]=1; wanted_output[i]=1;
} }

View File

@ -32,7 +32,7 @@ void forward_propagation(Network* network);
/* /*
* Propage en arrière le cnn * Propage en arrière le cnn
*/ */
void backward_propagation(Network* network, float wanted_number); void backward_propagation(Network* network, int wanted_number);
/* /*
* Met à 0 chaque valeur de l'input avec une probabilité de dropout % * Met à 0 chaque valeur de l'input avec une probabilité de dropout %
@ -57,6 +57,6 @@ float compute_cross_entropy_loss(float* output, float* wanted_output, int len);
/* /*
* On considère que la sortie voulue comporte 10 éléments * On considère que la sortie voulue comporte 10 éléments
*/ */
float* generate_wanted_output(float wanted_number); float* generate_wanted_output(int wanted_number, int size_output);
#endif #endif