SHOGUN  5.0.0
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules
MultitaskClusteredLogisticRegression.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 MULTITASKCLUSTEREDLOGISTICREGRESSION_H_
12 #define MULTITASKCLUSTEREDLOGISTICREGRESSION_H_
13 
14 #include <shogun/lib/config.h>
15 #ifdef USE_GPL_SHOGUN
16 
18 
19 namespace shogun
20 {
27 class CMultitaskClusteredLogisticRegression : public CMultitaskLogisticRegression
28 {
29 
30  public:
32 
33 
34  CMultitaskClusteredLogisticRegression();
35 
45  CMultitaskClusteredLogisticRegression(
46  float64_t rho1, float64_t rho2, CDotFeatures* training_data,
47  CBinaryLabels* training_labels, CTaskGroup* task_group,
48  int32_t num_clusters);
49 
51  virtual ~CMultitaskClusteredLogisticRegression();
52 
57  int32_t get_rho1() const;
58 
62  void set_rho1(float64_t rho1);
63 
66  int32_t get_rho2() const;
67 
71  void set_rho2(float64_t rho2);
72 
77  int32_t get_num_clusters() const;
78 
82  void set_num_clusters(int32_t num_clusters);
83 
88  virtual const char* get_name() const
89  {
90  return "MultitaskClusteredLogisticRegression";
91  }
92 
93  protected:
94 
99  virtual bool train_machine(CFeatures* data=NULL);
100 
105  virtual bool train_locked_implementation(SGVector<index_t>* tasks);
106 
107  protected:
108 
110  float64_t m_rho1;
111 
113  float64_t m_rho2;
114 
116  int32_t m_num_clusters;
117 };
118 }
119 #endif //USE_GPL_SHOGUN
120 #endif
121 
#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