SHOGUN  3.2.1
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
MulticlassTreeGuidedLogisticRegression.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 Sergey Lisitsyn
8  * Copyright (C) 2012 Sergey Lisitsyn
9  */
10 
11 #ifndef MULTICLASSTREEGUIDEDLOGISTICREGRESSION_H_
12 #define MULTICLASSTREEGUIDEDLOGISTICREGRESSION_H_
13 #include <shogun/lib/config.h>
14 #ifdef HAVE_EIGEN3
15 #include <shogun/lib/common.h>
19 
20 namespace shogun
21 {
22 
26 {
27  public:
29 
30 
32 
40 
43 
45  virtual const char* get_name() const
46  {
47  return "MulticlassTreeGuidedLogisticRegression";
48  }
49 
53  inline void set_z(float64_t z)
54  {
55  ASSERT(z>0)
56  m_z = z;
57  }
61  inline float64_t get_z() const { return m_z; }
62 
67  {
68  ASSERT(epsilon>0)
70  }
74  inline float64_t get_epsilon() const { return m_epsilon; }
75 
79  inline void set_max_iter(int32_t max_iter)
80  {
81  ASSERT(max_iter>0)
82  m_max_iter = max_iter;
83  }
87  inline int32_t get_max_iter() const { return m_max_iter; }
88 
92  inline void set_index_tree(CIndexBlockTree* index_tree)
93  {
94  SG_REF(index_tree);
96  m_index_tree = index_tree;
97  }
102  {
104  return m_index_tree;
105  }
106 
107 protected:
108 
110  virtual bool train_machine(CFeatures* data = NULL);
111 
112 private:
113 
115  void init_defaults();
116 
118  void register_parameters();
119 
120 protected:
121 
124 
127 
130 
132  int32_t m_max_iter;
133 
134 };
135 }
136 #endif /* HAVE_EIGEN3 */
137 #endif

SHOGUN Machine Learning Toolbox - Documentation