BayesF32.h 1.1 KB

12345678910111213141516171819202122232425262728293031
  1. #include "Test.h"
  2. #include "Pattern.h"
  3. class BayesF32:public Client::Suite
  4. {
  5. public:
  6. BayesF32(Testing::testID_t id);
  7. virtual void setUp(Testing::testID_t,std::vector<Testing::param_t>& paramsArgs,Client::PatternMgr *mgr);
  8. virtual void tearDown(Testing::testID_t,Client::PatternMgr *mgr);
  9. private:
  10. #include "BayesF32_decl.h"
  11. Client::Pattern<float32_t> input;
  12. Client::Pattern<float32_t> params;
  13. Client::Pattern<int16_t> dims;
  14. Client::LocalPattern<float32_t> outputProbas;
  15. Client::LocalPattern<int16_t> outputPredicts;
  16. // Reference patterns are not loaded when we are in dump mode
  17. Client::RefPattern<float32_t> probas;
  18. Client::RefPattern<int16_t> predicts;
  19. int nbPatterns,classNb,vecDim;
  20. const float32_t *theta;
  21. const float32_t *sigma;
  22. const float32_t *classPrior;
  23. float32_t epsilon;
  24. arm_gaussian_naive_bayes_instance_f32 bayes;
  25. };