BootstrapTree.hpp

Go to the documentation of this file.

#pragma once
 
#include "utils/Types.hpp"
#include "models/Tree.hpp"
 
#include <vector>
 
namespace ppforest2 {
  struct BootstrapTree : public Tree {
    using Tree::Tree;
    using Ptr = std::unique_ptr<BootstrapTree>;

    static Ptr train(TrainingSpec::Ptr const& training_spec,
                     types::FeatureMatrix const& x,
                     stats::GroupPartition const& group_spec,
                     stats::RNG& rng);
 
    std::vector<int> sample_indices;
 
    BootstrapTree(TreeNode::Ptr root, TrainingSpec::Ptr spec, std::vector<int> samp);

    std::vector<int> oob_indices(int n_total) const;

    types::ResponseVector predict_oob(types::FeatureMatrix const& x, std::vector<int> const& row_idx) const;
  };
}