SHOGUN  v3.0.0
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
LeastAngleRegression.h
Go to the documentation of this file.
1 /*
2  * This program is free software; you can redistribute it and/or modify
3  * it under the terms of the GNU General Public License as published by
4  * the Free Software Foundation; either version 3 of the License, or
5  * (at your option) any later version.
6  *
7  * Written (W) 2012 Chiyuan Zhang
8  * Copyright (C) 2012 Chiyuan Zhang
9  */
10 
11 #ifndef LEASTANGLEREGRESSION_H__
12 #define LEASTANGLEREGRESSION_H__
13 
14 #include <shogun/lib/config.h>
15 
16 #ifdef HAVE_LAPACK
17 #include <vector>
19 
20 namespace shogun
21 {
22 class CFeatures;
23 
73 {
74 public:
77 
82  CLeastAngleRegression(bool lasso=true);
83 
85  virtual ~CLeastAngleRegression();
86 
91  void set_max_non_zero(int32_t n)
92  {
93  m_max_nonz = n;
94  }
95 
98  int32_t get_max_non_zero() const
99  {
100  return m_max_nonz;
101  }
102 
108  {
109  m_max_l1_norm = norm;
110  }
111 
115  {
116  return m_max_l1_norm;
117  }
118 
123  void switch_w(int32_t num_variable)
124  {
125  if (w.vlen <= 0)
126  SG_ERROR("cannot swith estimator before training")
127  if (size_t(num_variable) >= m_beta_idx.size() || num_variable < 0)
128  SG_ERROR("cannot switch to an estimator of %d non-zero coefficients", num_variable)
129  if (w.vector == NULL)
131  std::copy(m_beta_path[m_beta_idx[num_variable]].begin(),
132  m_beta_path[m_beta_idx[num_variable]].end(), w.vector);
133  }
134 
143  int32_t get_path_size() const
144  {
145  return m_beta_idx.size();
146  }
147 
157  SGVector<float64_t> get_w(int32_t num_var)
158  {
159  return SGVector<float64_t>(&m_beta_path[m_beta_idx[num_var]][0], w.vlen, false);
160  }
161 
167  {
168  return CT_LARS;
169  }
170 
172  virtual const char* get_name() const { return "LeastAngleRegression"; }
173 
174 protected:
175  virtual bool train_machine(CFeatures* data=NULL);
176 
177 private:
178  void activate_variable(int32_t v)
179  {
180  m_num_active++;
181  m_active_set.push_back(v);
182  m_is_active[v] = true;
183  }
184  void deactivate_variable(int32_t v_idx)
185  {
186  m_num_active--;
187  m_is_active[m_active_set[v_idx]] = false;
188  m_active_set.erase(m_active_set.begin() + v_idx);
189  }
190 
191  SGMatrix<float64_t> cholesky_insert(SGMatrix<float64_t> X, SGMatrix<float64_t> R, int32_t i_max_corr);
192  SGMatrix<float64_t> cholesky_delete(SGMatrix<float64_t> R, int32_t i_kick);
193 
194 
195  bool m_lasso;
196 
197  int32_t m_max_nonz;
198  float64_t m_max_l1_norm;
199 
200  std::vector<std::vector<float64_t> > m_beta_path;
201  std::vector<int32_t> m_beta_idx;
202 
203  std::vector<int32_t> m_active_set;
204  std::vector<bool> m_is_active;
205  int32_t m_num_active;
206 }; // class LARS
207 
208 } // namespace shogun
209 
210 #endif // HAVE_LAPACK
211 #endif // LEASTANGLEREGRESSION_H__

SHOGUN Machine Learning Toolbox - Documentation