ProjectionPursuit.hpp

Go to the documentation of this file.

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

namespace ppforest2 {
  struct NodeContext;
}
 
namespace ppforest2::pp {
  class ProjectionPursuit : public Strategy<ProjectionPursuit> {
  public:
    struct Result {
      ppforest2::pp::Projector projector;
      types::Feature index_value = 0;
    };

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

  ProjectionPursuit::Ptr pda(float lambda);
}