Grouping.hpp

Go to the documentation of this file.

#pragma once
 
#include "models/strategies/Strategy.hpp"
#include "stats/GroupPartition.hpp"
#include "stats/Stats.hpp"
#include "utils/Types.hpp"

namespace ppforest2 {
  struct NodeContext;
}
 
namespace ppforest2::grouping {
  class Grouping : public Strategy<Grouping> {
  public:
    stats::GroupPartition init(types::OutcomeVector const& y) const;

    stats::GroupPartition init(types::GroupIdVector const& y) const {
      return init(types::OutcomeVector(y.cast<types::Outcome>()));
    }

    void split(NodeContext& ctx, stats::RNG& rng) const;
 
  protected:
    virtual stats::GroupPartition compute_init(types::OutcomeVector const& y) const = 0;

    virtual void compute(NodeContext& ctx, stats::RNG& rng) const = 0;
  };

  Grouping::Ptr by_label();

  Grouping::Ptr by_cutpoint();
}