DPx.hpp

Go to the documentation of this file.

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

 
class DPx : public DP {
 
protected:
    std::vector<std::shared_ptr<Module>> modules;
 
public:
    DPx(Data &d, Params &p, const std::vector<std::shared_ptr<Module>> &mods) : DP(d, p), 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) 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 { return; };
};