Packages

c

org.apache.spark.ml.tuning

ParamGridBuilder

class ParamGridBuilder extends AnyRef

Builder for a param grid used in grid search-based model selection.

Annotations
@Since( "1.2.0" )
Source
ParamGridBuilder.scala
Linear Supertypes
AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. ParamGridBuilder
  2. AnyRef
  3. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new ParamGridBuilder()
    Annotations
    @Since( "1.2.0" )

Value Members

  1. def addGrid(param: BooleanParam): ParamGridBuilder.this.type

    Adds a boolean param with true and false.

    Adds a boolean param with true and false.

    Annotations
    @Since( "1.2.0" )
  2. def addGrid(param: LongParam, values: Array[Long]): ParamGridBuilder.this.type

    Adds a long param with multiple values.

    Adds a long param with multiple values.

    Annotations
    @Since( "1.2.0" )
  3. def addGrid(param: FloatParam, values: Array[Float]): ParamGridBuilder.this.type

    Adds a float param with multiple values.

    Adds a float param with multiple values.

    Annotations
    @Since( "1.2.0" )
  4. def addGrid(param: IntParam, values: Array[Int]): ParamGridBuilder.this.type

    Adds an int param with multiple values.

    Adds an int param with multiple values.

    Annotations
    @Since( "1.2.0" )
  5. def addGrid(param: DoubleParam, values: Array[Double]): ParamGridBuilder.this.type

    Adds a double param with multiple values.

    Adds a double param with multiple values.

    Annotations
    @Since( "1.2.0" )
  6. def addGrid[T](param: Param[T], values: Iterable[T]): ParamGridBuilder.this.type

    Adds a param with multiple values (overwrites if the input param exists).

    Adds a param with multiple values (overwrites if the input param exists).

    Annotations
    @Since( "1.2.0" )
  7. def baseOn(paramPairs: ParamPair[_]*): ParamGridBuilder.this.type

    Sets the given parameters in this grid to fixed values.

    Sets the given parameters in this grid to fixed values.

    Annotations
    @Since( "1.2.0" ) @varargs()
  8. def baseOn(paramMap: ParamMap): ParamGridBuilder.this.type

    Sets the given parameters in this grid to fixed values.

    Sets the given parameters in this grid to fixed values.

    Annotations
    @Since( "1.2.0" )
  9. def build(): Array[ParamMap]

    Builds and returns all combinations of parameters specified by the param grid.

    Builds and returns all combinations of parameters specified by the param grid.

    Annotations
    @Since( "1.2.0" )