90 lines
2.6 KiB
C++
Executable File
90 lines
2.6 KiB
C++
Executable File
#include "BayesF16.h"
|
|
#include <stdio.h>
|
|
#include "Error.h"
|
|
#include "Test.h"
|
|
|
|
#define REL_ERROR ((float16_t)3e-3)
|
|
|
|
void BayesF16::test_gaussian_naive_bayes_predict_f16()
|
|
{
|
|
const float16_t *inp = input.ptr();
|
|
|
|
float16_t *bufp = outputProbas.ptr();
|
|
float16_t *tempp = temp.ptr();
|
|
int16_t *p = outputPredicts.ptr();
|
|
|
|
|
|
for(int i=0; i < this->nbPatterns ; i ++)
|
|
{
|
|
*p = arm_gaussian_naive_bayes_predict_f16(&bayes,
|
|
inp,
|
|
bufp,tempp);
|
|
|
|
inp += this->vecDim;
|
|
bufp += this->classNb;
|
|
p++;
|
|
}
|
|
|
|
ASSERT_REL_ERROR(outputProbas,probas,REL_ERROR);
|
|
ASSERT_EQ(outputPredicts,predicts);
|
|
}
|
|
|
|
|
|
void BayesF16::setUp(Testing::testID_t id,std::vector<Testing::param_t>& paramsArgs,Client::PatternMgr *mgr)
|
|
{
|
|
|
|
|
|
(void)paramsArgs;
|
|
|
|
switch(id)
|
|
{
|
|
case BayesF16::TEST_GAUSSIAN_NAIVE_BAYES_PREDICT_F16_1:
|
|
|
|
|
|
input.reload(BayesF16::INPUTS1_F16_ID,mgr);
|
|
params.reload(BayesF16::PARAMS1_F16_ID,mgr);
|
|
dims.reload(BayesF16::DIMS1_S16_ID,mgr);
|
|
|
|
const int16_t *dimsp=dims.ptr();
|
|
const float16_t *paramsp = params.ptr();
|
|
|
|
this->nbPatterns=dimsp[0];
|
|
this->classNb=dimsp[1];
|
|
this->vecDim=dimsp[2];
|
|
|
|
this->theta=paramsp;
|
|
this->sigma=paramsp + (this->classNb * this->vecDim);
|
|
this->classPrior=paramsp + 2*(this->classNb * this->vecDim);
|
|
this->epsilon=paramsp[this->classNb + 2*(this->classNb * this->vecDim)];
|
|
//printf("%f %f %f\n",this->theta[0],this->sigma[0],this->classPrior[0]);
|
|
|
|
// Reference patterns are not loaded when we are in dump mode
|
|
probas.reload(BayesF16::PROBAS1_F16_ID,mgr);
|
|
predicts.reload(BayesF16::PREDICTS1_S16_ID,mgr);
|
|
|
|
outputProbas.create(this->nbPatterns*this->classNb,BayesF16::OUT_PROBA_F16_ID,mgr);
|
|
temp.create(this->nbPatterns*this->classNb,BayesF16::OUT_PROBA_F16_ID,mgr);
|
|
outputPredicts.create(this->nbPatterns,BayesF16::OUT_PREDICT_S16_ID,mgr);
|
|
|
|
bayes.vectorDimension=this->vecDim;
|
|
bayes.numberOfClasses=this->classNb;
|
|
bayes.theta=this->theta;
|
|
bayes.sigma=this->sigma;
|
|
bayes.classPriors=this->classPrior;
|
|
bayes.epsilon=this->epsilon;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
void BayesF16::tearDown(Testing::testID_t id,Client::PatternMgr *mgr)
|
|
{
|
|
(void)id;
|
|
outputProbas.dump(mgr);
|
|
outputPredicts.dump(mgr);
|
|
}
|