SHOGUN  v3.0.0
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
MulticlassOneVsRestStrategy.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 Chiyuan Zhang
8  * Written (W) 2013 Shell Hu and Heiko Strathmann
9  * Copyright (C) 2012 Chiyuan Zhang
10  */
11 
12 #ifndef MULTICLASSONEVSRESTSTRATEGY_H__
13 #define MULTICLASSONEVSRESTSTRATEGY_H__
14 
16 
17 namespace shogun
18 {
19 
32 {
33 public:
36 
41 
44 
46  virtual void train_start(CMulticlassLabels *orig_labels, CBinaryLabels *train_labels)
47  {
48  CMulticlassStrategy::train_start(orig_labels, train_labels);
49  }
50 
52  virtual bool train_has_more()
53  {
54  return m_train_iter < m_num_classes;
55  }
56 
61 
65  virtual int32_t decide_label(SGVector<float64_t> outputs);
66 
71  virtual SGVector<index_t> decide_label_multiple_output(SGVector<float64_t> outputs, int32_t n_outputs);
72 
75  virtual int32_t get_num_machines()
76  {
77  return m_num_classes;
78  }
79 
81  virtual const char* get_name() const
82  {
83  return "MulticlassOneVsRestStrategy";
84  };
85 
89  virtual void rescale_outputs(SGVector<float64_t> outputs);
90 
97  virtual void rescale_outputs(SGVector<float64_t> outputs,
98  const SGVector<float64_t> As, const SGVector<float64_t> Bs);
99 
100 protected:
105 
112  const SGVector<float64_t> As, const SGVector<float64_t> Bs);
113 
114 };
115 
116 } // namespace shogun
117 
118 #endif /* end of include guard: MULTICLASSONEVSRESTSTRATEGY_H__ */
119 

SHOGUN Machine Learning Toolbox - Documentation