00001 /* 00002 * This program is free software; you can redistribute it and/or modify 00003 * it under the terms of the GNU General Public License as published by 00004 * the Free Software Foundation; either version 3 of the License, or 00005 * (at your option) any later version. 00006 * 00007 * Written (W) 1999-2009 Soeren Sonnenburg 00008 * Copyright (C) 1999-2009 Fraunhofer Institute FIRST and Max-Planck-Society 00009 */ 00010 00011 #ifndef _LIBSVM_H___ 00012 #define _LIBSVM_H___ 00013 00014 #include <shogun/lib/common.h> 00015 #include <shogun/classifier/svm/SVM.h> 00016 #include <shogun/lib/external/shogun_libsvm.h> 00017 00018 namespace shogun 00019 { 00020 #ifndef DOXYGEN_SHOULD_SKIP_THIS 00021 enum LIBSVM_SOLVER_TYPE 00022 { 00023 LIBSVM_C_SVC = 1, 00024 LIBSVM_NU_SVC = 2 00025 }; 00026 #endif 00027 00028 class CLibSVM : public CSVM 00029 { 00030 public: 00032 CLibSVM(LIBSVM_SOLVER_TYPE st=LIBSVM_C_SVC); 00033 00040 CLibSVM(float64_t C, CKernel* k, CLabels* lab); 00041 00042 virtual ~CLibSVM(); 00043 00048 virtual EMachineType get_classifier_type() { return CT_LIBSVM; } 00049 00051 virtual const char* get_name() const { return "LibSVM"; } 00052 00053 protected: 00062 virtual bool train_machine(CFeatures* data=NULL); 00063 00064 protected: 00066 svm_problem problem; 00068 svm_parameter param; 00070 struct svm_model* model; 00071 00073 LIBSVM_SOLVER_TYPE solver_type; 00074 }; 00075 } 00076 #endif