Merging requests
[mldemos:mldemos.git] / _AlgorithmsPlugins / KernelMethods / regressorSVR.h
1 /*********************************************************************\r
2 MLDemos: A User-Friendly visualization toolkit for machine learning\r
3 Copyright (C) 2010  Basilio Noris\r
4 Contact: mldemos@b4silio.com\r
5 \r
6 This library is free software; you can redistribute it and/or\r
7 modify it under the terms of the GNU Lesser General Public License,\r
8 version 3 as published by the Free Software Foundation.\r
9 \r
10 This library is distributed in the hope that it will be useful, but\r
11 WITHOUT ANY WARRANTY; without even the implied warranty of\r
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU\r
13 Lesser General Public License for more details.\r
14 \r
15 You should have received a copy of the GNU Lesser General Public\r
16 License along with this library; if not, write to the Free\r
17 Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.\r
18 *********************************************************************/\r
19 #ifndef _REGRESSOR_SVR_H_\r
20 #define _REGRESSOR_SVR_H_\r
21 \r
22 #include <vector>\r
23 #include <regressor.h>\r
24 #include "svm.h"\r
25 \r
26 class RegressorSVR : public Regressor\r
27 {\r
28 private:\r
29         svm_model *svm;\r
30         svm_node *node;\r
31 public:\r
32         svm_parameter param;\r
33     bool bOptimize;\r
34 \r
35         RegressorSVR();\r
36         ~RegressorSVR();\r
37         void Train(std::vector< fvec > samples, ivec labels);\r
38         fvec Test( const fvec &sample);\r
39         fVec Test(const fVec &sample);\r
40     void Optimize(svm_problem *problem);\r
41     const char *GetInfoString();\r
42 \r
43         void SetParams(int svmType, float svmC, float svmP, u32 kernelType, float kernelParam);\r
44     svm_model *GetModel(){return svm;}\r
45 };\r
46 \r
47 #endif // _REGRESSOR_SVR_H_\r