ByCutpoint.hpp

Go to the documentation of this file.

#pragma once
 
#include "models/strategies/grouping/Grouping.hpp"
#include "models/strategies/Strategy.hpp"
 
namespace ppforest2::grouping {
  class ByCutpoint : public Grouping {
  public:
    static Grouping::Ptr from_json(nlohmann::json const& j);
 
    nlohmann::json to_json() const override;
    std::string display_name() const override { return "By cutpoint"; }
    std::set<types::Mode> supported_modes() const override { return {types::Mode::Regression}; }
 
    PPFOREST2_REGISTER_STRATEGY(Grouping, "by_cutpoint")
 
  protected:
    stats::GroupPartition compute_init(types::OutcomeVector const& y) const override;

    void compute(NodeContext& ctx, stats::RNG& rng) const override;
  };
}