NGGP.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "../params/NGGP-params.hpp"
#include "../samplers/U_sampler/U_sampler.hpp"
#include "../utils/Process.hpp"

class NGGP : public Process {
 
protected:
 
  const NGGP_params &params;

  U_sampler &U_sampler_method;


  std::random_device rd;

  mutable std::mt19937 gen;


  const double log_a = log(params.a);
 
public:
  NGGP(Data &d, NGGP_params &p, U_sampler &mh)
      : Process(d), params(p), gen(rd()), U_sampler_method(mh) {};


  [[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 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 { U_sampler_method.update_U(); };

};