Go to the documentation of this file.00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036 #ifndef DOXYGEN_SHOULD_SKIP_THIS
00037
00038 #ifndef _LIBSVM_H
00039 #define _LIBSVM_H
00040
00041 #include <shogun/kernel/Kernel.h>
00042
00043 namespace shogun
00044 {
00046 struct svm_node
00047 {
00049 int32_t index;
00050 };
00051
00053 struct svm_problem
00054 {
00055
00056
00057 public:
00058
00060 svm_problem() {
00061 y = NULL;
00062 x = NULL;
00063 C = NULL;
00064 pv = NULL;
00065 }
00066
00067
00069 int32_t l;
00071 float64_t *y;
00073 struct svm_node **x;
00075 float64_t *C;
00077 float64_t *pv;
00078
00079 };
00080
00081 enum { C_SVC=1, NU_SVC=2, NU_MULTICLASS_SVC=3, ONE_CLASS=4, EPSILON_SVR=5, NU_SVR=6 };
00082 enum { LINEAR, POLY, RBF, SIGMOID, PRECOMPUTED };
00083
00085 struct svm_parameter
00086 {
00088 int32_t svm_type;
00090 int32_t kernel_type;
00092 shogun::CKernel* kernel;
00094 int32_t degree;
00096 float64_t gamma;
00098 float64_t coef0;
00099
00100
00102 float64_t cache_size;
00104 float64_t max_train_time;
00106 float64_t eps;
00108 float64_t C;
00110 int32_t nr_class;
00112 int32_t nr_weight;
00114 int32_t *weight_label;
00116 float64_t* weight;
00118 float64_t nu;
00120 float64_t p;
00122 int32_t shrinking;
00124 bool use_bias;
00125 };
00126
00128 struct svm_model
00129 {
00131 svm_parameter param;
00133 int32_t nr_class;
00135 int32_t l;
00137 svm_node **SV;
00139 float64_t **sv_coef;
00141 float64_t *rho;
00143 float64_t *normwcw;
00144
00145
00146
00148 int32_t *label;
00152 int32_t *nSV;
00153
00157 int32_t free_sv;
00159 float64_t objective;
00160 };
00161
00162
00163
00164 struct svm_model *svm_train(
00165 const struct svm_problem *prob, const struct svm_parameter *param);
00166
00167 float64_t svm_predict(const struct svm_model *model, const struct svm_node *x);
00168
00169 void svm_destroy_model(struct svm_model *model);
00170
00171 const char *svm_check_parameter(const struct svm_problem *prob, const struct svm_parameter *param);
00172 }
00173 #endif
00174
00175 #endif // DOXYGEN_SHOULD_SKIP_THIS