DPW.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "../utils/Process.hpp"
#include "Eigen/src/Core/Matrix.h"

class DPW : public Process {
 
public:
  DPW(Data &d, Params &p) : Process(d, p) {};


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