NGGPx.hpp

Go to the documentation of this file.

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

 
class NGGPx : public NGGP {
 
protected:
  std::vector<std::shared_ptr<Module>> modules;
 
public:
  NGGPx(Data &d, NGGP_params &p, U_sampler &U_sam,
        const std::vector<std::shared_ptr<Module>> &mods)
      : NGGP(d, p, U_sam), 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 = 0) 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 { NGGP::U_sampler_method.update_U(); };

};