commit 51d9f3cb2703f96622e0db753f44dd1c942bba6f
author Kishinami <dangorogoro0802@gmail.com> Thu Sep 24 14:32:04 2020 +0900
committer Christophe Favergeon <48906714+christophe0606@users.noreply.github.com> Fri Sep 25 07:28:59 2020 +0200
tree 9b52db95326996dcd01a879b800a0989182bad41
parent 7c7d2e6510da2c7d63663340b0a02d653198011c

CMSIS-DSP: Modified the arm_bayes_example

CMSIS/DSP/Examples/ARM/arm_bayes_example/arm_bayes_example_f32.c

diff --git a/CMSIS/DSP/Examples/ARM/arm_bayes_example/arm_bayes_example_f32.c b/CMSIS/DSP/Examples/ARM/arm_bayes_example/arm_bayes_example_f32.c
index 207ae6a..6470257 100755
--- a/CMSIS/DSP/Examples/ARM/arm_bayes_example/arm_bayes_example_f32.c
+++ b/CMSIS/DSP/Examples/ARM/arm_bayes_example/arm_bayes_example_f32.c
@@ -104,9 +104,9 @@
   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 @@
   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 @@
   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);