SHOGUN  3.2.1
 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 
13 #include <shogun/lib/config.h>
14 
16 
17 namespace shogun
18 {
26 {
27 
28  public:
30 
31 
33 
44  float64_t rho1, float64_t rho2, CDotFeatures* training_data,
45  CBinaryLabels* training_labels, CTaskGroup* task_group,
46  int32_t num_clusters);
47 
50 
55  int32_t get_rho1() const;
56 
60  void set_rho1(float64_t rho1);
61 
64  int32_t get_rho2() const;
65 
69  void set_rho2(float64_t rho2);
70 
75  int32_t get_num_clusters() const;
76 
80  void set_num_clusters(int32_t num_clusters);
81 
86  virtual const char* get_name() const
87  {
88  return "MultitaskClusteredLogisticRegression";
89  }
90 
91  protected:
92 
97  virtual bool train_machine(CFeatures* data=NULL);
98 
103  virtual bool train_locked_implementation(SGVector<index_t>* tasks);
104 
105  protected:
106 
109 
112 
114  int32_t m_num_clusters;
115 };
116 }
117 #endif

SHOGUN Machine Learning Toolbox - Documentation