LeafStrategy.hpp

Go to the documentation of this file.

#pragma once
 
#include "models/TreeNode.hpp"
#include "models/strategies/Strategy.hpp"
#include "stats/Stats.hpp"
 
namespace ppforest2 {
  struct NodeContext;
}

namespace ppforest2::leaf {
  class LeafStrategy : public Strategy<LeafStrategy> {
  public:
    TreeNode::Ptr create_leaf(NodeContext const& ctx, stats::RNG& rng) const { return compute(ctx, rng); }
 
  protected:
    virtual TreeNode::Ptr compute(NodeContext const& ctx, stats::RNG& rng) const = 0;
  };

  LeafStrategy::Ptr majority_vote();

  LeafStrategy::Ptr mean_response();
}