Skip to content

Commit 6e0b3d7

Browse files
wjurrxin
authored andcommitted
[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 <[email protected]> Closes #11252 from wjur/wjur/docs_multiclass.
1 parent a153e41 commit 6e0b3d7

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

mllib/src/main/scala/org/apache/spark/mllib/tree/impurity/Entropy.scala

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,7 @@ import org.apache.spark.annotation.{DeveloperApi, Experimental, Since}
2121

2222
/**
2323
* :: Experimental ::
24-
* Class for calculating [[http://en.wikipedia.org/wiki/Binary_entropy_function entropy]] during
25-
* binary classification.
24+
* Class for calculating entropy during multiclass classification.
2625
*/
2726
@Since("1.0.0")
2827
@Experimental

mllib/src/main/scala/org/apache/spark/mllib/tree/impurity/Gini.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ import org.apache.spark.annotation.{DeveloperApi, Experimental, Since}
2323
* :: Experimental ::
2424
* Class for calculating the
2525
* [[http://en.wikipedia.org/wiki/Decision_tree_learning#Gini_impurity Gini impurity]]
26-
* during binary classification.
26+
* during multiclass classification.
2727
*/
2828
@Since("1.0.0")
2929
@Experimental

0 commit comments

Comments
 (0)