PPPDAStrategy.hpp

Go to the documentation of this file.

#pragma once
 
#include "utils/Types.hpp"
#include "utils/JsonValidation.hpp"
#include "models/PPStrategy.hpp"
#include "models/Strategy.hpp"
 
namespace ppforest2::pp {
  struct PPPDAStrategy : public PPStrategy {
    explicit PPPDAStrategy(float lambda);
 
    void to_json(nlohmann::json& j) const override;
    std::string display_name() const override { return lambda == 0 ? "LDA" : "PDA"; }
 
    types::Feature index(types::FeatureMatrix const& x,
                         stats::GroupPartition const& group_spec,
                         Projector const& projector) const override;
 
    PPResult optimize(types::FeatureMatrix const& x, stats::GroupPartition const& group_spec) const override;
 
    static PPStrategy::Ptr from_json(nlohmann::json const& j);
 
    PPFOREST2_REGISTER_STRATEGY(PPStrategy, "pda")
 
  private:
    float const lambda;
  };

  PPStrategy::Ptr pda(float lambda);
}