svm.h

Go to the documentation of this file.
00001 /*
00002 Copyright (c) 2000-2008 Chih-Chung Chang and Chih-Jen Lin
00003 All rights reserved.
00004 
00005 Redistribution and use in source and binary forms, with or without
00006 modification, are permitted provided that the following conditions
00007 are met:
00008 
00009 1. Redistributions of source code must retain the above copyright
00010 notice, this list of conditions and the following disclaimer.
00011 
00012 2. Redistributions in binary form must reproduce the above copyright
00013 notice, this list of conditions and the following disclaimer in the
00014 documentation and/or other materials provided with the distribution.
00015 
00016 3. Neither name of copyright holders nor the names of its contributors
00017 may be used to endorse or promote products derived from this software
00018 without specific prior written permission.
00019 
00020 
00021 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
00022 ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
00023 LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
00024 A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR
00025 CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
00026 EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
00027 PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
00028 PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
00029 LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
00030 NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
00031 SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
00032 
00033 */
00034 #ifndef _LIBSVM_H
00035 #define _LIBSVM_H
00036 
00037 #define LIBSVM_VERSION 286
00038 
00039 namespace celeste
00040 {
00041 #ifdef __cplusplus
00042 extern "C" {
00043 #endif
00044 
00045 struct svm_node
00046 {
00047         int index;
00048         double value;
00049 };
00050 
00051 struct svm_problem
00052 {
00053         int l;
00054         double *y;
00055         struct svm_node **x;
00056 };
00057 
00058 enum { C_SVC, NU_SVC, ONE_CLASS, EPSILON_SVR, NU_SVR }; /* svm_type */
00059 enum { LINEAR, POLY, RBF, SIGMOID, PRECOMPUTED }; /* kernel_type */
00060 
00061 struct svm_parameter
00062 {
00063         int svm_type;
00064         int kernel_type;
00065         int degree;     /* for poly */
00066         double gamma;   /* for poly/rbf/sigmoid */
00067         double coef0;   /* for poly/sigmoid */
00068 
00069         /* these are for training only */
00070         double cache_size; /* in MB */
00071         double eps;     /* stopping criteria */
00072         double C;       /* for C_SVC, EPSILON_SVR and NU_SVR */
00073         int nr_weight;          /* for C_SVC */
00074         int *weight_label;      /* for C_SVC */
00075         double* weight;         /* for C_SVC */
00076         double nu;      /* for NU_SVC, ONE_CLASS, and NU_SVR */
00077         double p;       /* for EPSILON_SVR */
00078         int shrinking;  /* use the shrinking heuristics */
00079         int probability; /* do probability estimates */
00080 };
00081 
00082 struct svm_model *svm_train(const struct svm_problem *prob, const struct svm_parameter *param);
00083 void svm_cross_validation(const struct svm_problem *prob, const struct svm_parameter *param, int nr_fold, double *target);
00084 
00085 int svm_save_model(const char *model_file_name, const struct svm_model *model);
00086 struct svm_model *svm_load_model(const char *model_file_name);
00087 
00088 int svm_get_svm_type(const struct svm_model *model);
00089 int svm_get_nr_class(const struct svm_model *model);
00090 void svm_get_labels(const struct svm_model *model, int *label);
00091 double svm_get_svr_probability(const struct svm_model *model);
00092 
00093 void svm_predict_values(const struct svm_model *model, const struct svm_node *x, double* dec_values);
00094 double svm_predict(const struct svm_model *model, const struct svm_node *x);
00095 double svm_predict_probability(const struct svm_model *model, const struct svm_node *x, double* prob_estimates);
00096 
00097 void svm_destroy_model(struct svm_model *model);
00098 void svm_destroy_param(struct svm_parameter *param);
00099 
00100 const char *svm_check_parameter(const struct svm_problem *prob, const struct svm_parameter *param);
00101 int svm_check_probability_model(const struct svm_model *model);
00102 
00103 #ifdef __cplusplus
00104 }
00105 #endif
00106 };  //namespace
00107 #endif /* _LIBSVM_H */

Generated on 5 Dec 2014 for Hugintrunk by  doxygen 1.4.7