Forest.hpp

Go to the documentation of this file.

#pragma once
 
#include "models/Bagged.hpp"
#include "models/Model.hpp"
#include "models/Tree.hpp"
#include "stats/Stats.hpp"
 
#include <memory>
#include <vector>
 
namespace ppforest2 {
  class Forest : public Model {
  public:
    using Ptr = std::unique_ptr<Forest>;
    using Model::predict;
 
    using FeatureMatrix = types::FeatureMatrix;
    using FeatureVector = types::FeatureVector;
    using OutcomeVector = types::OutcomeVector;
    using Outcome       = types::Outcome;
    using RNG           = stats::RNG;

    std::vector<BaggedTree::Ptr> trees;

    static Ptr train(TrainingSpec const& spec, FeatureMatrix const& x, OutcomeVector const& y);

    Outcome predict(FeatureVector const& x) const override = 0;

    void accept(Model::Visitor& visitor) const override = 0;

    void add_tree(BaggedTree::Ptr tree);
 
    bool operator==(Forest const& other) const;
    bool operator!=(Forest const& other) const;
 
  protected:
    Forest() = default;
    explicit Forest(TrainingSpec::Ptr spec) { training_spec = std::move(spec); };

    virtual BaggedTree::Ptr train_tree(FeatureMatrix const& x, OutcomeVector const& y, RNG& rng) const = 0;

    void build_trees(types::FeatureMatrix const& x, types::OutcomeVector const& y);
  };
}