SHOGUN  v2.0.0
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
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 #ifndef MULTITASKCLUSTEREDLOGISTICREGRESSION_H_
11 #define MULTITASKCLUSTEREDLOGISTICREGRESSION_H_
12 
14 
15 namespace shogun
16 {
24 {
25 
26  public:
28 
29 
31 
42  float64_t rho1, float64_t rho2, CDotFeatures* training_data,
43  CBinaryLabels* training_labels, CTaskGroup* task_group,
44  int32_t num_clusters);
45 
48 
53  int32_t get_rho1() const;
54 
58  void set_rho1(float64_t rho1);
59 
62  int32_t get_rho2() const;
63 
67  void set_rho2(float64_t rho2);
68 
73  int32_t get_num_clusters() const;
74 
78  void set_num_clusters(int32_t num_clusters);
79 
84  virtual const char* get_name() const
85  {
86  return "MultitaskClusteredLogisticRegression";
87  }
88 
89  protected:
90 
95  virtual bool train_machine(CFeatures* data=NULL);
96 
101  virtual bool train_locked_implementation(SGVector<index_t>* tasks);
102 
103  protected:
104 
107 
110 
112  int32_t m_num_clusters;
113 };
114 }
115 #endif

SHOGUN Machine Learning Toolbox - Documentation