DP.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "../params/DP-params.hpp"
#include "../utils/Process.hpp"

class DP : public Process {
  const DP_params &params;
  const double log_a;
 
public:
  DP(Data &d, DP_params &p)
      : Process(d), params(p), log_a(std::log(params.a)) {};


  [[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 { return; };
};