binary_covariate_module.hpp

Go to the documentation of this file.

#pragma once

 
#include "../../utils/Data.hpp"
#include "../../utils/Module.hpp"

 
class BinaryCovariatesModule : public Module {
protected:
 
    const double beta_prior_alpha = 1.0; 
    const double beta_prior_beta = 1.0;  

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


 
    const double log_beta_prior =
        std::lgamma(beta_prior_alpha) + std::lgamma(beta_prior_beta) -
        std::lgamma(beta_prior_alpha + beta_prior_beta); 
 
public:
    BinaryCovariatesModule(const Data &data_, const Eigen::VectorXi binary_covariate, double beta_prior_alpha_,
                           double beta_prior_beta_, const Eigen::VectorXi *old_alloc_provider = nullptr,
                           const std::unordered_map<int, std::vector<int>> *old_cluster_members_provider_ = nullptr)
        : beta_prior_alpha(beta_prior_alpha_), beta_prior_beta(beta_prior_beta_), data(data_),
          binary_covariate_data(binary_covariate), Module(old_alloc_provider, old_cluster_members_provider_) {}


    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));

};