diff --git a/src/mlpack/methods/decision_tree/information_gain.hpp b/src/mlpack/methods/decision_tree/information_gain.hpp index 6126a41918..cc48600fa6 100644 --- a/src/mlpack/methods/decision_tree/information_gain.hpp +++ b/src/mlpack/methods/decision_tree/information_gain.hpp @@ -26,7 +26,7 @@ class InformationGain { public: /** - * Evaluate the Gini impurity given a vector of class weight counts. + * Evaluate the information gain given a vector of class weight counts. */ template static double EvaluatePtr(const CountType* counts,