SHOGUN  3.2.1
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
BalancedConditionalProbabilityTree.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  * Copyright (C) 2012 Chiyuan Zhang
9  */
10 
11 #ifndef BALANCEDCONDITIONALPROBABILITYTREE_H__
12 #define BALANCEDCONDITIONALPROBABILITYTREE_H__
13 
14 #include <shogun/lib/config.h>
15 
17 
18 namespace shogun
19 {
20 
38 {
39 public:
42 
45 
47  virtual const char* get_name() const { return "BalancedConditionalProbabilityTree"; }
48 
50  void set_alpha(float64_t alpha);
51 
53  float64_t get_alpha() const { return m_alpha; }
54 
55 protected:
61  virtual bool which_subtree(bnode_t *node, SGVector<float32_t> ex);
62 
63 private:
68  int32_t tree_depth(bnode_t *node);
69 
71  float64_t m_alpha;
72 };
73 
74 } /* shogun */
75 
76 #endif /* end of include guard: BALANCEDCONDITIONALPROBABILITYTREE_H__ */
77 

SHOGUN Machine Learning Toolbox - Documentation