SHOGUN  3.2.1
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Machine.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) 1999-2009 Soeren Sonnenburg
8  * Written (W) 2011-2012 Heiko Strathmann
9  * Copyright (C) 1999-2009 Fraunhofer Institute FIRST and Max-Planck-Society
10  */
11 
12 #ifndef _MACHINE_H__
13 #define _MACHINE_H__
14 
15 #include <shogun/lib/config.h>
16 
17 #include <shogun/lib/common.h>
18 #include <shogun/base/SGObject.h>
25 
26 namespace shogun
27 {
28 
29 class CFeatures;
30 class CLabels;
31 
34 {
35  CT_NONE = 0,
36  CT_LIGHT = 10,
38  CT_LIBSVM = 20,
41  CT_MPD = 50,
42  CT_GPBT = 60,
46  CT_LDA = 100,
47  CT_LPM = 110,
48  CT_LPBOOST = 120,
49  CT_KNN = 130,
50  CT_SVMLIN=140,
52  CT_GNPPSVM = 160,
53  CT_GMNPSVM = 170,
54  CT_SVMPERF = 200,
55  CT_LIBSVR = 210,
56  CT_SVRLIGHT = 220,
57  CT_LIBLINEAR = 230,
58  CT_KMEANS = 240,
60  CT_SVMOCAS = 260,
61  CT_WDSVMOCAS = 270,
62  CT_SVMSGD = 280,
68  CT_DASVM = 340,
69  CT_LARANK = 350,
73  CT_SGDQN = 390,
77  CT_QDA = 430,
78  CT_NEWTONSVM = 440,
80  CT_LARS = 460,
86  CT_CCSOSVM = 520,
91  CT_BAGGING = 570,
92  CT_FWSOSVM = 580,
93  CT_BCFWSOSVM = 590,
95 };
96 
99 {
107 };
108 
111 {
118 };
119 
120 #define MACHINE_PROBLEM_TYPE(PT) \
121  \
124  virtual EProblemType get_machine_problem_type() const { return PT; }
125 
143 class CMachine : public CSGObject
144 {
145  public:
147  CMachine();
148 
150  virtual ~CMachine();
151 
161  virtual bool train(CFeatures* data=NULL);
162 
169  virtual CLabels* apply(CFeatures* data=NULL);
170 
172  virtual CBinaryLabels* apply_binary(CFeatures* data=NULL);
174  virtual CRegressionLabels* apply_regression(CFeatures* data=NULL);
176  virtual CMulticlassLabels* apply_multiclass(CFeatures* data=NULL);
178  virtual CStructuredLabels* apply_structured(CFeatures* data=NULL);
180  virtual CLatentLabels* apply_latent(CFeatures* data=NULL);
181 
186  virtual void set_labels(CLabels* lab);
187 
192  virtual CLabels* get_labels();
193 
199 
205 
211 
216  void set_solver_type(ESolverType st);
217 
223 
229  virtual void set_store_model_features(bool store_model);
230 
239  virtual bool train_locked(SGVector<index_t> indices)
240  {
241  SG_ERROR("train_locked(SGVector<index_t>) is not yet implemented "
242  "for %s\n", get_name());
243  return false;
244  }
247  virtual float64_t apply_one(int32_t i)
248  {
250  return 0.0;
251  }
252 
258  virtual CLabels* apply_locked(SGVector<index_t> indices);
259 
262  SGVector<index_t> indices);
265  SGVector<index_t> indices);
268  SGVector<index_t> indices);
271  SGVector<index_t> indices);
274  SGVector<index_t> indices);
275 
284  virtual void data_lock(CLabels* labs, CFeatures* features);
287  virtual void post_lock(CLabels* labs, CFeatures* features) { };
288 
290  virtual void data_unlock();
293  virtual bool supports_locking() const { return false; }
296  bool is_data_locked() const { return m_data_locked; }
299  virtual EProblemType get_machine_problem_type() const
300  {
302  return PT_BINARY;
303  }
304 
305  virtual const char* get_name() const { return "Machine"; }
306 
307  protected:
318  virtual bool train_machine(CFeatures* data=NULL)
319  {
320  SG_ERROR("train_machine is not yet implemented for %s!\n",
321  get_name());
322  return false;
323  }
324 
335  virtual void store_model_features()
336  {
337  SG_ERROR("Model storage and therefore unlocked Cross-Validation and"
338  " Model-Selection is not supported for %s. Locked may"
339  " work though.\n", get_name());
340  }
341 
348  virtual bool is_label_valid(CLabels *lab) const
349  {
350  return true;
351  }
354  virtual bool train_require_labels() const { return true; }
355 
356  protected:
361  CLabels* m_labels;
370  bool m_data_locked;
371 };
372 }
373 #endif // _MACHINE_H__

SHOGUN Machine Learning Toolbox - Documentation