DPx.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "../utils/Module.hpp"
#include "DP.hpp"
#include <memory>

 
class DPx : public DP {
 
protected:
  std::vector<std::shared_ptr<Module>> modules;
 
public:
  DPx(Data &d, DP_params &p, const std::vector<std::shared_ptr<Module>> &mods)
      : DP(d, p), modules(mods) {
    for (auto &mod : modules) {
      mod->set_old_allocations_provider(&this->old_allocations_view());
      mod->set_old_cluster_members_provider(&this->old_cluster_members_view());
    }
  }


  [[nodiscard]] double gibbs_prior_existing_cluster(int cls_idx,
                                                    int obs_idx) const override;

  [[nodiscard]] Eigen::VectorXd
  gibbs_prior_existing_clusters(int obs_idx) const override;

  [[nodiscard]] double gibbs_prior_new_cluster() const override;

  [[nodiscard]] double gibbs_prior_new_cluster_obs(int obs_idx) const override;



  [[nodiscard]] double prior_ratio_split(int ci, int cj) const override;

  [[nodiscard]] double prior_ratio_merge(int size_old_ci,
                                         int size_old_cj) const override;

  [[nodiscard]] double prior_ratio_shuffle(int size_old_ci, int size_old_cj,
                                           int ci, int cj) const override;


  void update_params() override { return; };
};