diff options
author | Wojciech Jurczyk <wojciech.jurczyk@codilime.com> | 2016-06-15 15:58:42 -0700 |
---|---|---|
committer | Reynold Xin <rxin@databricks.com> | 2016-06-15 15:58:42 -0700 |
commit | 6e0b3d795c572a9df589e148c0e55e4eeba7946c (patch) | |
tree | d36fd2381cd0f244dfb1097421b61bdef87a7713 | |
parent | a153e41c08b635b1b2a48b680d203c148deebc67 (diff) | |
download | spark-6e0b3d795c572a9df589e148c0e55e4eeba7946c.tar.gz spark-6e0b3d795c572a9df589e148c0e55e4eeba7946c.tar.bz2 spark-6e0b3d795c572a9df589e148c0e55e4eeba7946c.zip |
[DOCS] Fix Gini and Entropy scaladocs in context of multiclass classification
The PR changes outdated scaladocs for Gini and Entropy classes. Since PR #886 Spark supports multiclass classification, but the docs tell only about binary classification.
Author: Wojciech Jurczyk <wojciech.jurczyk@codilime.com>
Closes #11252 from wjur/wjur/docs_multiclass.
-rw-r--r-- | mllib/src/main/scala/org/apache/spark/mllib/tree/impurity/Entropy.scala | 3 | ||||
-rw-r--r-- | mllib/src/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 ff7700d2d1..3a731f45d6 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 58dc79b739..7730c0a8c1 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 |