clusters.h

Go to the documentation of this file.

 
#pragma once
 
#include <Eigen/SparseCore>
#include <vector>
 
namespace slope {
 
class Clusters
{
public:
  Clusters() = default;
 
  Clusters(const Eigen::VectorXd& beta);
 
  std::size_t size() { return c.size(); }
 
  std::vector<int>::iterator begin(const int i);
 
  std::vector<int>::iterator end(const int i);
 
  std::vector<int>::const_iterator cbegin(const int i) const;
 
  std::vector<int>::const_iterator cend(const int i) const;
 
  int cluster_size(const int i) const;
 
  int pointer(const int i) const;
 
  int size() const;
 
  double coeff(const int i) const;
 
  void setCoeff(const int i, const double x);
 
  const std::vector<double>& coeffs() const;
 
  const std::vector<int>& indices() const;
 
  const std::vector<int>& pointers() const;
 
  void update(const int old_index, const int new_index, const double c_new);
 
  void update(const Eigen::VectorXd& beta);
 
  std::vector<std::vector<int>> getClusters() const;
 
private:
  std::vector<double> c;  
  std::vector<int> c_ind; 
  std::vector<int>
    c_ptr; 
  int p;   
  void reorder(const int old_index, const int new_index);
 
  void merge(const int old_index, const int new_index);
};
 
Eigen::SparseMatrix<int>
patternMatrix(const Eigen::VectorXd& beta);
 
} // namespace slope