From d08cfc7c24e1f6cd24260f5d36cec9f99c7cf951 Mon Sep 17 00:00:00 2001 From: Wojciech Jurczyk Date: Thu, 18 Feb 2016 13:42:48 +0100 Subject: [PATCH] [Docs] Fix Gini and Entropy scaladocs --- .../scala/org/apache/spark/mllib/tree/impurity/Entropy.scala | 3 +-- .../main/scala/org/apache/spark/mllib/tree/impurity/Gini.scala | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/mllib/src/main/scala/org/apache/spark/mllib/tree/impurity/Entropy.scala b/mllib/src/main/scala/org/apache/spark/mllib/tree/impurity/Entropy.scala index 73df6b054a8ce..872e7c0c406c6 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/tree/impurity/Entropy.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/tree/impurity/Entropy.scala @@ -21,8 +21,7 @@ import org.apache.spark.annotation.{DeveloperApi, Experimental, Since} /** * :: Experimental :: - * Class for calculating [[http://en.wikipedia.org/wiki/Binary_entropy_function entropy]] during - * binary classification. + * Class for calculating entropy during multiclass classification. */ @Since("1.0.0") @Experimental diff --git a/mllib/src/main/scala/org/apache/spark/mllib/tree/impurity/Gini.scala b/mllib/src/main/scala/org/apache/spark/mllib/tree/impurity/Gini.scala index f21845b21a802..9afe392d700dd 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/tree/impurity/Gini.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/tree/impurity/Gini.scala @@ -23,7 +23,7 @@ import org.apache.spark.annotation.{DeveloperApi, Experimental, Since} * :: Experimental :: * Class for calculating the * [[http://en.wikipedia.org/wiki/Decision_tree_learning#Gini_impurity Gini impurity]] - * during binary classification. + * during multiclass classification. */ @Since("1.0.0") @Experimental