- #ifndef SVM_H
- #define SVM_H
- #include <vector>
- #include "libsvm.h"
- class svm
- {
- public:
- svm();
- ~svm();
- void train(float* features, int* labels, int feature_size, int feature_num);
- int classify(float* feature);
- private:
- void clean_up();
- svm_problem problem;
- svm_parameter param;
- svm_node* x_space;
- svm_model* model;
- float scale;
- int feature_length;
- };
- #endif