Added documentation

This commit is contained in:
Rishabh Garg
2021-10-20 13:36:32 +05:30
parent c6350424e8
commit cc092e330c
2 changed files with 5 additions and 0 deletions
@@ -487,7 +487,10 @@ class DecisionTree :
union
{
//! Stores the type of dimension on which the split is done for internal
//! nodes of the tree.
size_t dimensionType;
//! Stores the majority class for leaf nodes of the tree.
size_t majorityClass;
};
/**
@@ -453,7 +453,9 @@ class DecisionTreeRegressor :
union
{
//! Stores the split point for internal nodes of the tree.
double splitPoint;
//! Stores the prediction value for leaf nodes of the tree.
double prediction;
};