svm_alg.hh

Go to the documentation of this file.

#ifndef _SVM_ALGHH_
#define _SVM_ALGHH_

#include <openmodeller/om.hh>

#include "svm.h"

/**************************************************/
/************* Support Vector Machines ************/

class SvmAlgorithm : public AlgorithmImpl
{
public:

  SvmAlgorithm();

  ~SvmAlgorithm();

  int needNormalization();
  int initialize();
  int iterate();
  int done() const;

  Scalar getValue( const Sample& x ) const;
  int getConvergence( Scalar * const val ) const;

protected:

  void _getNode( svm_node * node, const Sample& sample ) const;

  void _getConfiguration( ConfigurationPtr& ) const;

  void _setConfiguration( const ConstConfigurationPtr& );

  bool _done;

  int _num_layers;

  svm_parameter _svm_parameter;
  
  svm_problem _svm_problem;

  svm_model * _svm_model;

  // Index of presence class in SVM arrays. No need to serialize it since it is
  // detected at run time. -1 means "to be determined".
  int _presence_index;
};


#endif