NGGPx.hpp

Go to the documentation of this file.

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

 
class NGGPx : public NGGP {
 
protected:
    std::vector<std::shared_ptr<Module>> modules;
 
public:
    NGGPx(Data &d, Params &p, U_sampler &U_sam, const std::vector<std::shared_ptr<Module>> &mods)
        : NGGP(d, p, U_sam), 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 = 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 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 { NGGP::U_sampler_method.update_U(); };

};