mitkLibSVMClassifier.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef mitkLibSVMClassifier_h
#define mitkLibSVMClassifier_h
 
#include <MitkCLLibSVMExports.h>
 
#include <mitkAbstractClassifier.h>
 
struct svm_parameter;
struct svm_problem;
struct svm_node;
struct svm_model;
 
namespace mitk
{
  class MITKCLLIBSVM_EXPORT LibSVMClassifier : public AbstractClassifier
  {
  public:
 
    mitkClassMacro(LibSVMClassifier,AbstractClassifier);
 
    itkFactorylessNewMacro(Self);
 
    itkCloneMacro(Self);
 
    LibSVMClassifier();
    ~LibSVMClassifier() override;
 
    void Train(const Eigen::MatrixXd &X, const Eigen::MatrixXi &Y) override;
    Eigen::MatrixXi Predict(const Eigen::MatrixXd &X) override;
 
    bool SupportsPointWiseWeight() override{return true;}
    bool SupportsPointWiseProbability() override{return false;}
 
    void PrintParameter(std::ostream & str);
    void ConvertParameter();
 
    void SetSvmType(int val);
    void SetProbability(int val);
    void SetShrinking(int val);
    void SetNrWeight(int val);
    void SetNu(double val);
    void SetP(double val);
    void SetEps(double val);
    void SetC(double val);
    void SetCacheSize(double val);
    void SetKernelType(int val);
    void SetDegree(int val);
    void SetGamma(double val);
    void SetCoef0(double val);
 
  private:
 
    void ReadXValues(svm_problem * problem, svm_node** xSpace, const Eigen::MatrixXd &X);
    void ReadYValues(svm_problem * problem, const Eigen::MatrixXi &Y);
    void ReadWValues(svm_problem * problem);
 
    svm_model* m_Model;
    svm_parameter * m_Parameter;
 
  };
}
 
#endif