categorical_covariate_module.hpp

Go to the documentation of this file.

#pragma once

 
#include "../../utils/Data.hpp"
#include "../../utils/Module.hpp"
#include <numeric>

 
class CategoricalCovariatesModule : public Module {
protected:
 
    const std::vector<double> prior_alpha; 

    const Data &data;
 
    const Eigen::VectorXi categorical_covariate_data; 


 
    const double alpha_0 = std::accumulate(prior_alpha.begin(), prior_alpha.end(), 0.0); 
    const double lgamma_alpha_0 = std::lgamma(alpha_0); 
 
    double prod_lgamma_prior = 0; 

 
public:
    CategoricalCovariatesModule(
        const Data &data_, const Eigen::VectorXi categorical_covariate, std::vector<double> prior_alpha_,
        const Eigen::VectorXi *old_alloc_provider = nullptr,
        const std::unordered_map<int, std::vector<int>> *old_cluster_members_provider_ = nullptr)
        : prior_alpha(prior_alpha_), data(data_), categorical_covariate_data(categorical_covariate),
          Module(old_alloc_provider, old_cluster_members_provider_) {
 
        for (const double &alpha : prior_alpha)
            prod_lgamma_prior += std::lgamma(alpha);
    }


    double compute_similarity_cls(int cls_idx, bool old_allo = false) const override __attribute__((hot));

    double compute_similarity_obs(int obs_idx, int cls_idx) const override __attribute__((hot));

    Eigen::VectorXd compute_similarity_obs(int obs_idx) const override __attribute__((hot));

};