rf_alg.hh

Go to the documentation of this file.

#ifndef _RF_ALGHH_
#define _RF_ALGHH_

#include <vector>

#include <openmodeller/om.hh>
#include <openmodeller/Random.hh>

#include "librf/instance_set.h"
#include "librf/tree.h"

/*****************************************/
/************* Random Forests ************/

class RfAlgorithm : public AlgorithmImpl
{
public:

  RfAlgorithm();

  ~RfAlgorithm();

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

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

protected:

  void _sampleToLine( Sample sample, stringstream& ss ) const;

  void _getConfiguration( ConfigurationPtr& ) const;

  void _setConfiguration( const ConstConfigurationPtr& );

  bool _done;

  int _num_trees;

  int _k;

  bool _initialized;

  Random _rand;

  librf::InstanceSet* _set; // only used for model creation

  vector<int> _class_weights;

  vector<librf::Tree*> _trees;
};


#endif