From b9cffb6fa9c82b442bbf12e6b2b3f6fc0c5bcb27 Mon Sep 17 00:00:00 2001
From: Holden Karau <holden@pigscanfly.ca>
Date: Fri, 31 Jul 2015 11:33:33 -0700
Subject: [PATCH] Update the labels param to have the metadata note

---
 .../main/scala/org/apache/spark/ml/feature/StringIndexer.scala | 3 ++-
 1 file changed, 2 insertions(+), 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
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