Github user hhbyyh commented on a diff in the pull request: https://github.com/apache/spark/pull/19599#discussion_r156556136 --- Diff: mllib/src/main/scala/org/apache/spark/ml/param/params.scala --- @@ -435,6 +435,43 @@ class BooleanParam(parent: String, name: String, doc: String) // No need for isV } } +/** + * :: DeveloperApi :: + * Specialized version of `Param[String]` for Java. + */ +@DeveloperApi +class StringParam(parent: Params, name: String, doc: String, isValid: String => Boolean) + extends Param[String](parent, name, doc, isValid) { + + private var options: Option[Array[String]] = None + + def this(parent: Params, name: String, doc: String) = + this(parent, name, doc, ParamValidators.alwaysTrue) + + /** construct a StringParam with limited options (case-insensitive) */ + def this(parent: Params, name: String, doc: String, options: Array[String]) = { + this(parent, name, doc + s" Supported options (case-insensitive): ${options.mkString(", ")}.", --- End diff -- It's currently not. Do you mean we should test the `doc` correctly include all the options? or there's any special case that should be covered.
--- --------------------------------------------------------------------- To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org For additional commands, e-mail: reviews-h...@spark.apache.org