spatial_module_continuous.hpp

Go to the documentation of this file.

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

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

  void neighbor_cache_compute();
 
  const Data
      &data_module; 
  const Eigen::MatrixXd W; 
  const double spatial_weight =
      1.0; 
 
public:
  SpatialModuleContinuous(const Data &data_, const Eigen::MatrixXd 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;
};