1 /*
2 * Copyright (c) 2021 Stephanos Ioannidis <root@stephanos.io>
3 * Copyright (C) 2010-2021 ARM Limited or its affiliates. All rights reserved.
4 *
5 * SPDX-License-Identifier: Apache-2.0
6 */
7
8 #include <zephyr/ztest.h>
9 #include <zephyr/kernel.h>
10 #include <stdlib.h>
11 #include <arm_math.h>
12 #include "../../common/test_common.h"
13
14 #include "f32.pat"
15
16 #define REL_ERROR_THRESH (5.0e-6)
17
18 ZTEST_SUITE(bayes_f32, NULL, NULL, NULL, NULL, NULL);
19
ZTEST(bayes_f32,test_gaussian_naive_bayes_predict_f32)20 ZTEST(bayes_f32, test_gaussian_naive_bayes_predict_f32)
21 {
22 arm_gaussian_naive_bayes_instance_f32 inst;
23
24 size_t index;
25 const uint16_t pattern_count = in_dims[0];
26 const uint16_t class_count = in_dims[1];
27 const uint16_t vec_dims = in_dims[2];
28
29 const float32_t *params = (const float32_t *)in_param;
30 const float32_t *input = (const float32_t *)in_val;
31 float32_t *output_probs_buf, *output_probs;
32 uint16_t *output_preds_buf, *output_preds;
33 float32_t *temp;
34
35 /* Initialise instance */
36 inst.vectorDimension = vec_dims;
37 inst.numberOfClasses = class_count;
38 inst.theta = params;
39 inst.sigma = params + (class_count * vec_dims);
40 inst.classPriors = params + (2 * class_count * vec_dims);
41 inst.epsilon = params[class_count + (2 * class_count * vec_dims)];
42
43 /* Allocate output buffers */
44 output_probs_buf =
45 malloc(pattern_count * class_count * sizeof(float32_t));
46 zassert_not_null(output_probs_buf, ASSERT_MSG_BUFFER_ALLOC_FAILED);
47
48 output_preds_buf =
49 malloc(pattern_count * sizeof(uint16_t));
50 zassert_not_null(output_preds_buf, ASSERT_MSG_BUFFER_ALLOC_FAILED);
51
52 output_probs = output_probs_buf;
53 output_preds = output_preds_buf;
54
55 temp = malloc(pattern_count * class_count * sizeof(float32_t));
56 zassert_not_null(temp, ASSERT_MSG_BUFFER_ALLOC_FAILED);
57
58 /* Enumerate patterns */
59 for (index = 0; index < pattern_count; index++) {
60 /* Run test function */
61 *output_preds =
62 arm_gaussian_naive_bayes_predict_f32(
63 &inst, input, output_probs, temp);
64
65 /* Increment pointers */
66 input += vec_dims;
67 output_probs += class_count;
68 output_preds++;
69 }
70
71 /* Validate output */
72 zassert_true(
73 test_rel_error_f32(pattern_count, output_probs_buf,
74 (float32_t *)ref_prob, REL_ERROR_THRESH),
75 ASSERT_MSG_REL_ERROR_LIMIT_EXCEED);
76
77 zassert_true(
78 test_equal_q15(pattern_count, output_preds_buf, ref_pred),
79 ASSERT_MSG_INCORRECT_COMP_RESULT);
80
81 /* Free output buffers */
82 free(output_probs_buf);
83 free(output_preds_buf);
84 }
85
