DRStrategy.hpp

Go to the documentation of this file.

#pragma once
 
#include "models/Projector.hpp"
#include "models/Strategy.hpp"
#include "stats/GroupPartition.hpp"
#include "stats/Stats.hpp"
#include "utils/Invariant.hpp"
#include "utils/Types.hpp"
 
#include <algorithm>
#include <vector>

namespace ppforest2::dr {
  struct DRResult {
    std::vector<int> const selected_cols;
    size_t const original_size;
 
    DRResult(std::vector<int> const& selected_cols, size_t const original_size)
        : selected_cols(selected_cols)
        , original_size(original_size) {}

    pp::Projector expand(pp::Projector const& reduced_vector) const {
      invariant(reduced_vector.size() == selected_cols.size(),
                "Reduced vector size must match number of selected variables");
 
      pp::Projector full_vector = pp::Projector::Zero(original_size);
 
      for (size_t i = 0; i < selected_cols.size(); ++i) {
        full_vector(selected_cols[i]) = reduced_vector(i);
      }
 
      return full_vector;
    }
  };

  struct DRStrategy : public Strategy<DRStrategy> {
    virtual DRResult
    select(types::FeatureMatrix const& x, stats::GroupPartition const& group_spec, stats::RNG& rng) const = 0;

    DRResult operator()(types::FeatureMatrix const& x, stats::GroupPartition const& group_spec, stats::RNG& rng) const {
      return select(x, group_spec, rng);
    }
  };
}