SHOGUN  3.2.1
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
GaussianNaiveBayes.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) 2011 Sergey Lisitsyn
8  * Copyright (C) 2011 Berlin Institute of Technology and Max-Planck-Society
9  */
10 
11 #ifndef GAUSSIANNAIVEBAYES_H_
12 #define GAUSSIANNAIVEBAYES_H_
13 
14 #include <shogun/lib/config.h>
15 
19 
20 namespace shogun {
21 
22 class CLabels;
23 class CDotFeatures;
24 class CFeatures;
25 
38 {
39 
40 public:
42 
43 
47 
52  CGaussianNaiveBayes(CFeatures* train_examples, CLabels* train_labels);
53 
57  virtual ~CGaussianNaiveBayes();
58 
62  virtual void set_features(CFeatures* features);
63 
67  virtual CFeatures* get_features();
68 
73  virtual CMulticlassLabels* apply_multiclass(CFeatures* data=NULL);
74 
79  virtual float64_t apply_one(int32_t idx);
80 
84  virtual const char* get_name() const { return "GaussianNaiveBayes"; };
85 
90 
91 protected:
92 
97  virtual bool train_machine(CFeatures* data=NULL);
98 
99 protected:
100 
103 
105  int32_t m_min_label;
106 
108  int32_t m_num_classes;
109 
111  int32_t m_dim;
112 
115 
118 
121 
124 };
125 
126 }
127 
128 #endif /* GAUSSIANNAIVEBAYES_H_ */

SHOGUN Machine Learning Toolbox - Documentation