From 4f80b10161051132494b0cdb64d5363b14f91b95 Mon Sep 17 00:00:00 2001 From: Kishinami Date: Thu, 24 Sep 2020 14:32:04 +0900 Subject: [PATCH] CMSIS-DSP: Modified the arm_bayes_example --- .../ARM/arm_bayes_example/arm_bayes_example_f32.c | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/Examples/ARM/arm_bayes_example/arm_bayes_example_f32.c b/Examples/ARM/arm_bayes_example/arm_bayes_example_f32.c index 207ae6a1..64702573 100755 --- a/Examples/ARM/arm_bayes_example/arm_bayes_example_f32.c +++ b/Examples/ARM/arm_bayes_example/arm_bayes_example_f32.c @@ -104,9 +104,9 @@ int32_t main(void) in[0] = 1.5f; in[1] = 1.0f; - arm_gaussian_naive_bayes_predict_f32(&S, in, result); + index = arm_gaussian_naive_bayes_predict_f32(&S, in, result); - arm_max_f32(result, NB_OF_CLASSES, &maxProba, &index); + maxProba = result[index]; #if defined(SEMIHOSTING) printf("Class = %d\n", index); @@ -115,9 +115,9 @@ int32_t main(void) in[0] = -1.5f; in[1] = 1.0f; - arm_gaussian_naive_bayes_predict_f32(&S, in, result); + index = arm_gaussian_naive_bayes_predict_f32(&S, in, result); - arm_max_f32(result, NB_OF_CLASSES, &maxProba, &index); + maxProba = result[index]; #if defined(SEMIHOSTING) printf("Class = %d\n", index); @@ -126,9 +126,9 @@ int32_t main(void) in[0] = 0.0f; in[1] = -3.0f; - arm_gaussian_naive_bayes_predict_f32(&S, in, result); + index = arm_gaussian_naive_bayes_predict_f32(&S, in, result); - arm_max_f32(result, NB_OF_CLASSES, &maxProba, &index); + maxProba = result[index]; #if defined(SEMIHOSTING) printf("Class = %d\n", index);