Update the labels param to have the metadata note

apache · Jul 31, 2015 · b9cffb6 · b9cffb6
1 parent 6a38edb
commit b9cffb6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mllib/src/main/scala/org/apache/spark/ml/feature/StringIndexer.scala b/mllib/src/main/scala/org/apache/spark/ml/feature/StringIndexer.scala
@@ -201,7 +201,8 @@ class StringIndexerInverse private[ml] (
    * metadata is read for labels.
    * @group param
    */
-  final val labels: StringArrayParam = new StringArrayParam(this, "labels", "array of labels")
+  final val labels: StringArrayParam = new StringArrayParam(this, "labels",
+    "array of labels, if not provided metadata from inputCol is used instead.")
   setDefault(labels, Array.empty[String])
 
   /** @group getParam