spatial_module.hpp

Go to the documentation of this file.

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

class SpatialModule : public Module {
protected:
    std::vector<std::vector<int>> neighbor_cache;

    void neighbor_cache_compute();
 
    const Data &data_module;           
    const Eigen::MatrixXi W;           
    const double spatial_weight = 1.0; 
 
public:
    SpatialModule(const Data &data_, const Eigen::MatrixXi W_, double spatial_coeff,
                  const Eigen::VectorXi *old_alloc_provider = nullptr,
                  const std::unordered_map<int, std::vector<int>> *old_cluster_members_provider_ = nullptr)
        : W(W_), data_module(data_), spatial_weight(spatial_coeff),
          Module(old_alloc_provider, old_cluster_members_provider_) {
 
        neighbor_cache_compute();
    }


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

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

    double compute_similarity_obs(int obs_idx, int cls_idx) const override;
};