diff options
author | Rekha Joshi <rekhajoshm@gmail.com> | 2015-07-17 20:02:05 -0700 |
---|---|---|
committer | Xiangrui Meng <meng@databricks.com> | 2015-07-17 20:02:05 -0700 |
commit | 1017908205b7690dc0b0ed4753b36fab5641f7ac (patch) | |
tree | 2133788f57a3e6c8d2561ec01d2e2b08d1c6f1ad | |
parent | 34a889db857f8752a0a78dcedec75ac6cd6cd48d (diff) | |
download | spark-1017908205b7690dc0b0ed4753b36fab5641f7ac.tar.gz spark-1017908205b7690dc0b0ed4753b36fab5641f7ac.tar.bz2 spark-1017908205b7690dc0b0ed4753b36fab5641f7ac.zip |
[SPARK-9118] [ML] Implement IntArrayParam in mllib
Implement IntArrayParam in mllib
Author: Rekha Joshi <rekhajoshm@gmail.com>
Author: Joshi <rekhajoshm@gmail.com>
Closes #7481 from rekhajoshm/SPARK-9118 and squashes the following commits:
d3b1766 [Joshi] Implement IntArrayParam
0be142d [Rekha Joshi] Merge pull request #3 from apache/master
106fd8e [Rekha Joshi] Merge pull request #2 from apache/master
e3677c9 [Rekha Joshi] Merge pull request #1 from apache/master
-rw-r--r-- | mllib/src/main/scala/org/apache/spark/ml/param/params.scala | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/ml/param/params.scala b/mllib/src/main/scala/org/apache/spark/ml/param/params.scala index d034d7ec6b..824efa5ed4 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/param/params.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/param/params.scala @@ -296,6 +296,22 @@ class DoubleArrayParam(parent: Params, name: String, doc: String, isValid: Array } /** + * :: DeveloperApi :: + * Specialized version of [[Param[Array[Int]]]] for Java. + */ +@DeveloperApi +class IntArrayParam(parent: Params, name: String, doc: String, isValid: Array[Int] => Boolean) + extends Param[Array[Int]](parent, name, doc, isValid) { + + def this(parent: Params, name: String, doc: String) = + this(parent, name, doc, ParamValidators.alwaysTrue) + + /** Creates a param pair with a [[java.util.List]] of values (for Java and Python). */ + def w(value: java.util.List[java.lang.Integer]): ParamPair[Array[Int]] = + w(value.asScala.map(_.asInstanceOf[Int]).toArray) +} + +/** * :: Experimental :: * A param and its value. */ |