source: trunk/src/SVM.h @ 30

// $Id: SVM.h 30 2004-01-16 17:40:51Z peter $

#ifndef CS_CPP_TOOLS_SVM_H
#define CS_CPP_TOOLS_SVM_H

// C++ tools include
/////////////////////
#include "vector.h"
#include "matrix.h"

// Standard C++ includes
////////////////////////


namespace thep_cpp_tools 
{
  
  class SVM
  {
  
  public:
    /**
       Constructor taking the kernel matrix and the target vector as input
    */
    SVM(const thep_gsl_api::matrix&, const thep_gsl_api::vector&);
           
    /**
       Training the SVM using the SMO algorithm
    */
    void train();

    /**
       Function will return \f$\alpha\f$
    */
    inline thep_gsl_api::vector get_alpha() const;
   
      
  private:
    bool trained_;
    thep_gsl_api::matrix kernel_;
    thep_gsl_api::vector target_;
    thep_gsl_api::vector alpha_;
  }; 
  
    // class SVM
  thep_gsl_api::vector SVM::get_alpha() const  
  {
    return alpha_;
  }
}; // namespace thep_c++_tools

#endif