diff --git a/src/mlpack/methods/decision_stump/decision_stump.hpp b/src/mlpack/methods/decision_stump/decision_stump.hpp index 2cd4eed7d9..7f2b495124 100644 --- a/src/mlpack/methods/decision_stump/decision_stump.hpp +++ b/src/mlpack/methods/decision_stump/decision_stump.hpp @@ -115,7 +115,8 @@ class DecisionStump * @param predictedLabels Vector to store the predicted classes after * classifying test data. */ - mlpack_deprecated void Classify(const MatType& test, arma::Row& predictedLabels); + mlpack_deprecated void Classify(const MatType& test, + arma::Row& predictedLabels); //! Access the splitting dimension. size_t SplitDimension() const { return splitDimension; } diff --git a/src/mlpack/methods/decision_tree/decision_tree.hpp b/src/mlpack/methods/decision_tree/decision_tree.hpp index 48f5852b0d..9b3deef594 100644 --- a/src/mlpack/methods/decision_tree/decision_tree.hpp +++ b/src/mlpack/methods/decision_tree/decision_tree.hpp @@ -590,7 +590,8 @@ using DecisionStump = DecisionTree; /** - * Convenience typedef for decision tree (Three level decision trees). + * Convenience typedef for ID3decision stumps (single level decision trees + * made with ID3 algorithm). */ typedef DecisionTree