random_forest.h

Go to the documentation of this file.

#ifndef _RANDOM_FOREST_H_
#define _RANDOM_FOREST_H_

#include <vector>

using namespace std;

namespace librf {

class Instance;
class InstanceSet;
class Tree;
class RandomForest {
  public:
    RandomForest();
    RandomForest(const InstanceSet& set,
                 int num_trees,
                 int K,
                 int max_depth);
    ~RandomForest();
     // int predict(const Instance& c) const;
     // float predict_prob(const Instance& c) const;
     int predict(const InstanceSet& set, int instance_no) const;
     float predict_prob(const InstanceSet& set, int instance_no, int label) const;
     float testing_accuracy(const InstanceSet& testset) const;
     float training_accuracy() const;
     float oob_accuracy() const;
     void variable_importance(vector< pair<float, int> >* ranking,
                              unsigned int* seed) const;
     void read(istream& i);
     void write(ostream& o);
     void print() const;
  private:
    const InstanceSet& set_; // training data set
    vector<Tree*> trees_; // component trees in the forest
    int max_depth_; // maximum depth of trees
    int K_; // random vars to try per split
};

} // namespace
#endif

---