Forest.hpp

Go to the documentation of this file.

#pragma once
 
#include "models/Model.hpp"
#include "models/BootstrapTree.hpp"
 
#include <map>
#include <memory>
#include <numeric>
#include <thread>
#include <vector>
 
namespace ppforest2 {
  struct Forest : public Model {
    static Forest
    train(TrainingSpec const& training_spec, types::FeatureMatrix const& x, types::ResponseVector const& y);
 
    std::vector<BootstrapTree::Ptr> trees;
 
    Forest();
    explicit Forest(TrainingSpec::Ptr training_spec);

    types::Response predict(types::FeatureVector const& data) const override;

    types::ResponseVector predict(types::FeatureMatrix const& data) const override;

    types::FeatureMatrix predict(types::FeatureMatrix const& data, Proportions) const override;

    void add_tree(std::unique_ptr<BootstrapTree> tree);
 
    bool operator==(Forest const& other) const;
    bool operator!=(Forest const& other) const;
 
    void accept(Model::Visitor& visitor) const override;

    types::ResponseVector oob_predict(types::FeatureMatrix const& x) const;

    double oob_error(types::FeatureMatrix const& x, types::ResponseVector const& y) const;
  };
}