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 index a406454a7e6ab..e464f1dfc2399 100644 --- 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