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

View File

@ -32,7 +32,7 @@ void forward_propagation(Network* network);
/*
* 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 %
@ -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
*/
float* generate_wanted_output(float wanted_number);
float* generate_wanted_output(int wanted_number, int size_output);
#endif