SVMF16.h 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. #include "Test.h"
  2. #include "Pattern.h"
  3. #include "dsp/svm_functions_f16.h"
  4. class SVMF16:public Client::Suite
  5. {
  6. public:
  7. SVMF16(Testing::testID_t id);
  8. virtual void setUp(Testing::testID_t,std::vector<Testing::param_t>& params,Client::PatternMgr *mgr);
  9. virtual void tearDown(Testing::testID_t,Client::PatternMgr *mgr);
  10. private:
  11. #include "SVMF16_decl.h"
  12. Client::Pattern<float16_t> samples;
  13. Client::Pattern<int16_t> dims;
  14. Client::Pattern<float16_t> params;
  15. arm_svm_linear_instance_f16 linear;
  16. arm_svm_polynomial_instance_f16 poly;
  17. arm_svm_rbf_instance_f16 rbf;
  18. arm_svm_sigmoid_instance_f16 sigmoid;
  19. int vecDim,nbSupportVectors,nbTestSamples,degree;
  20. int32_t classes[2]={0,0};
  21. float16_t intercept;
  22. const float16_t *supportVectors;
  23. const float16_t *dualCoefs;
  24. float16_t coef0, gamma;
  25. float16_t *inp;
  26. int nbLinear=0,nbPoly=0,nbRBF=0,nbSigmoid=0;
  27. enum {
  28. LINEAR=1,
  29. POLY=2,
  30. RBF=3,
  31. SIGMOID=4
  32. } kind;
  33. };