SHOGUN  5.0.0
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules
MultitaskLogisticRegression.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  * Copyright (C) 2012 Sergey Lisitsyn
8  */
9 
10 
11 #ifndef MULTITASKLOGISTICREGRESSION_H_
12 #define MULTITASKLOGISTICREGRESSION_H_
13 
14 #include <shogun/lib/config.h>
15 #ifdef USE_GPL_SHOGUN
21 
22 
23 namespace shogun
24 {
33 class CMultitaskLogisticRegression : public CMultitaskLinearMachine
34 {
35 
36  public:
39 
40 
41  CMultitaskLogisticRegression();
42 
50  CMultitaskLogisticRegression(
51  float64_t z, CDotFeatures* training_data,
52  CBinaryLabels* training_labels, CTaskRelation* task_relation);
53 
55  virtual ~CMultitaskLogisticRegression();
56 
58  virtual const char* get_name() const
59  {
60  return "MultitaskLogisticRegression";
61  }
62 
64  int32_t get_max_iter() const;
66  float64_t get_q() const;
68  int32_t get_regularization() const;
70  int32_t get_termination() const;
72  float64_t get_tolerance() const;
74  float64_t get_z() const;
75 
77  void set_max_iter(int32_t max_iter);
79  void set_q(float64_t q);
81  void set_regularization(int32_t regularization);
83  void set_termination(int32_t termination);
85  void set_tolerance(float64_t tolerance);
87  void set_z(float64_t z);
88 
90  virtual float64_t apply_one(int32_t i);
91 
92  protected:
93 
95  virtual bool train_machine(CFeatures* data=NULL);
96 
98  virtual bool train_locked_implementation(SGVector<index_t>* tasks);
99 
100  private:
101 
103  void register_parameters();
104 
106  void initialize_parameters();
107 
108  protected:
109 
111  int32_t m_regularization;
112 
114  int32_t m_termination;
115 
117  int32_t m_max_iter;
118 
120  float64_t m_tolerance;
121 
123  float64_t m_q;
124 
126  float64_t m_z;
127 
128 };
129 }
130 #endif //USE_GPL_SHOGUN
131 #endif
132 
#define MACHINE_PROBLEM_TYPE(PT)
Definition: Machine.h:120
double float64_t
Definition: common.h:50
all of classes and functions are contained in the shogun namespace
Definition: class_list.h:18

SHOGUN Machine Learning Toolbox - Documentation