StopRule.hpp

Go to the documentation of this file.

#pragma once
 
#include "models/strategies/Strategy.hpp"
#include "stats/Stats.hpp"
 
#include <vector>

namespace ppforest2 {
  struct NodeContext;
}
 
namespace ppforest2::stop {
  class StopRule : public Strategy<StopRule> {
  public:
    bool should_stop(NodeContext const& ctx, stats::RNG& rng) const;
 
  protected:
    virtual bool compute(NodeContext const& ctx, stats::RNG& rng) const = 0;
  };

  StopRule::Ptr pure_node();

  StopRule::Ptr min_size(int n);

  StopRule::Ptr min_variance(types::Feature threshold);

  StopRule::Ptr max_depth(int n);

  StopRule::Ptr any(std::vector<StopRule::Ptr> rules);
}