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 //mitkLibSVMClassifier_h