NGGPW.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "../utils/Process.hpp"

class NGGPW : public Process {
 
private:

  double U = 1;

  double tau = params.tau;



  void update_U();

  double log_conditional_density_V(double v) const;



  std::random_device rd;

  mutable std::mt19937 gen;



  static constexpr double proposal_std = 0.5;



  const double a_over_sigma = params.a / params.sigma;

  const double tau_power_sigma = std::pow(params.tau, params.sigma);



  int accepted_U = 0;

 
public:
  NGGPW(Data &d, Params &p) : Process(d, p), gen(rd()) {};


  [[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;



  int get_neighbors_obs(int obs_idx, int cls_idx) const;

  Eigen::VectorXi get_neighbors_obs(int obs_idx) const;

  int get_neighbors_cls(int cls_idx, bool old_allo = false) const;



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



  double get_U() const { return U; }

  int get_accepted_U() const { return accepted_U; }

};