#include <regressiontree.h>
Inheritance diagram for CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >:
Public Member Functions | |
BinaryRegressionTree (const Vector< int > &DDomainSize, int CsplitDim, int RegDim) | |
virtual | ~BinaryRegressionTree (void) |
virtual int | InDim (void) |
Get the number of input dimensions. | |
virtual string | TypeName (void) |
Get the type of this object. | |
virtual void | Infer (void) |
Do the inference. | |
virtual void | Identify (void) |
Use the training data to learn a new structure. | |
virtual void | Prune (void) |
Prune the structure. | |
virtual int | SetOption (char *name, char *val) |
Set an option for the machine. | |
virtual void | SaveToStream (ostream &out) |
Output the structure data to a stream. | |
Protected Member Functions | |
void | PrintSizesTree (void) |
Protected Attributes | |
BinaryRegressionTreeNode< T_Distribution, T_Regressor, T_Splitter > * | root |
const Vector< int > & | dDomainSize |
list of discrete domain sizes | |
int | dsplitDim |
num of discrete variables | |
int | csplitDim |
num of continuous+split variables | |
int | regDim |
num of regression variables | |
int | emMaxIterations |
num of iterations to get convergence for EM | |
int | emRestarts |
num of restarts of EM to get a good initial starting point | |
int | min_no_datapoints |
the minimum number of datapoints in a node to split further | |
int | splitType |
type of split to be passed to splitter, splitter dependent | |
double | threshold |
threshold for considering a branch irrelevant | |
T_Distribution * | rootDistribution |
int | inferMaxNodeId |
|
Definition at line 92 of file regressiontree.h. |
|
Definition at line 105 of file regressiontree.h. |
|
Use the training data to learn a new structure.
Reimplemented from CLUS::Machine. Definition at line 139 of file regressiontree.h. |
|
Get the number of input dimensions.
Reimplemented from CLUS::Machine. Definition at line 111 of file regressiontree.h. |
|
Do the inference.
Reimplemented from CLUS::Machine. Definition at line 121 of file regressiontree.h. |
|
Definition at line 83 of file regressiontree.h. Referenced by CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::Identify(), and CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::Prune(). |
|
Prune the structure.
Reimplemented from CLUS::Machine. Definition at line 189 of file regressiontree.h. |
|
Output the structure data to a stream.
Reimplemented from CLUS::Machine. Definition at line 241 of file regressiontree.h. |
|
Set an option for the machine.
Reimplemented from CLUS::Machine. Definition at line 217 of file regressiontree.h. |
|
Get the type of this object.
Reimplemented from CLUS::Machine. Definition at line 116 of file regressiontree.h. Referenced by CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::SaveToStream(). |
|
|
list of discrete domain sizes
Definition at line 54 of file regressiontree.h. Referenced by CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::BinaryRegressionTree(), CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::Identify(), and CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::SaveToStream(). |
|
|
num of iterations to get convergence for EM
Definition at line 66 of file regressiontree.h. Referenced by CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::BinaryRegressionTree(), CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::Identify(), and CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::SetOption(). |
|
num of restarts of EM to get a good initial starting point
Definition at line 69 of file regressiontree.h. Referenced by CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::BinaryRegressionTree(), CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::Identify(), and CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::SetOption(). |
|
|
the minimum number of datapoints in a node to split further
Definition at line 72 of file regressiontree.h. Referenced by CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::BinaryRegressionTree(), CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::Identify(), and CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::SetOption(). |
|
|
|
|
type of split to be passed to splitter, splitter dependent
Definition at line 75 of file regressiontree.h. Referenced by CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::BinaryRegressionTree(), CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::Identify(), and CLUS::BinaryRegressionTree< T_Distribution, T_Regressor, T_Splitter >::SetOption(). |
|