From d3b1766de9e5e9e128245494c0e7211047c3edca Mon Sep 17 00:00:00 2001 From: Joshi Date: Fri, 17 Jul 2015 14:09:35 -0700 Subject: [PATCH] Implement IntArrayParam --- .../scala/org/apache/spark/ml/param/params.scala | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) 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 50c0d855066f8..c0fc1614af039 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 @@ -295,6 +295,22 @@ class DoubleArrayParam(parent: Params, name: String, doc: String, isValid: Array w(value.asScala.map(_.asInstanceOf[Double]).toArray) } +/** + * :: 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.