SVMF32.h 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. #include "Test.h"
  2. #include "Pattern.h"
  3. #include "dsp/svm_functions.h"
  4. class SVMF32:public Client::Suite
  5. {
  6. public:
  7. SVMF32(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 "SVMF32_decl.h"
  12. Client::Pattern<float32_t> samples;
  13. Client::Pattern<int16_t> dims;
  14. Client::Pattern<float32_t> params;
  15. Client::RefPattern<int32_t> ref;
  16. Client::LocalPattern<int32_t> output;
  17. arm_svm_linear_instance_f32 linear;
  18. arm_svm_polynomial_instance_f32 poly;
  19. arm_svm_rbf_instance_f32 rbf;
  20. arm_svm_sigmoid_instance_f32 sigmoid;
  21. int vecDim,nbSupportVectors,nbTestSamples,degree;
  22. int32_t classes[2]={0,0};
  23. float32_t intercept;
  24. const float32_t *supportVectors;
  25. const float32_t *dualCoefs;
  26. float32_t coef0, gamma;
  27. enum {
  28. LINEAR=1,
  29. POLY=2,
  30. RBF=3,
  31. SIGMOID=4
  32. } kind;
  33. };