From cc092e330c2bda43c4ecc4069b67a1c3a4223592 Mon Sep 17 00:00:00 2001
From: Rishabh Garg <56191449+RishabhGarg108@users.noreply.github.com>
Date: Wed, 20 Oct 2021 13:36:32 +0530
Subject: [PATCH] Added documentation

---
 src/mlpack/methods/decision_tree/decision_tree.hpp           | 3 +++
 src/mlpack/methods/decision_tree/decision_tree_regressor.hpp | 2 ++
 2 files changed, 5 insertions(+)

diff --git a/src/mlpack/methods/decision_tree/decision_tree.hpp b/src/mlpack/methods/decision_tree/decision_tree.hpp
index 79f31c0cff..eacf40ce7b 100644
--- a/src/mlpack/methods/decision_tree/decision_tree.hpp
+++ b/src/mlpack/methods/decision_tree/decision_tree.hpp
@@ -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;
   };
   /**
diff --git a/src/mlpack/methods/decision_tree/decision_tree_regressor.hpp b/src/mlpack/methods/decision_tree/decision_tree_regressor.hpp
index c0754388ba..a4e6231c16 100644
--- a/src/mlpack/methods/decision_tree/decision_tree_regressor.hpp
+++ b/src/mlpack/methods/decision_tree/decision_tree_regressor.hpp
@@ -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;
   };