keystoneml.nodes.learning

KernelRidgeRegression

class KernelRidgeRegression[T] extends LabelEstimator[T, DenseVector[Double], DenseVector[Double]]

Solves a kernel ridge regression problem of the form (K(x, x) + \lambda * I) * W = Y using Gauss-Seidel based Block Coordinate Descent.

The function K is specified by the kernel generator and this class uses the dual formulation of the ridge regression objective to improve numerical stability.

Linear Supertypes
LabelEstimator[T, DenseVector[Double], DenseVector[Double]], EstimatorOperator, Serializable, Serializable, Operator, AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. KernelRidgeRegression
  2. LabelEstimator
  3. EstimatorOperator
  4. Serializable
  5. Serializable
  6. Operator
  7. AnyRef
  8. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new KernelRidgeRegression(kernelGenerator: KernelGenerator[T], lambda: Double, blockSize: Int, numEpochs: Int, blockPermuter: Option[Long] = scala.None, blocksBeforeCheckpoint: Int = 25)(implicit arg0: ClassTag[T])

    kernelGenerator

    kernel function to apply to create kernel matrix

    lambda

    L2 regularization value

    blockSize

    number of columns in each block of BCD

    numEpochs

    number of epochs of BCD to run

    blockPermuter

    seed used for permuting column blocks in BCD

    blocksBeforeCheckpoint

    frequency at which intermediate data should be checkpointed

Value Members

  1. final def !=(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  2. final def !=(arg0: Any): Boolean

    Definition Classes
    Any
  3. final def ##(): Int

    Definition Classes
    AnyRef → Any
  4. final def ==(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  5. final def ==(arg0: Any): Boolean

    Definition Classes
    Any
  6. final def asInstanceOf[T0]: T0

    Definition Classes
    Any
  7. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  8. final def eq(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  9. def equals(arg0: Any): Boolean

    Definition Classes
    AnyRef → Any
  10. def execute(deps: Seq[Expression]): TransformerExpression

    Definition Classes
    EstimatorOperator → Operator
  11. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  12. def fit(data: RDD[T], labels: RDD[DenseVector[Double]]): KernelBlockLinearMapper[T]

    The type-safe method that ML developers need to implement when writing new Estimators.

    The type-safe method that ML developers need to implement when writing new Estimators.

    data

    The estimator's training data.

    labels

    The estimator's training labels

    returns

    A new transformer

    Definition Classes
    KernelRidgeRegressionLabelEstimator
  13. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any
  14. def hashCode(): Int

    Definition Classes
    AnyRef → Any
  15. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  16. def label: String

    Definition Classes
    Operator
  17. final def ne(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  18. final def notify(): Unit

    Definition Classes
    AnyRef
  19. final def notifyAll(): Unit

    Definition Classes
    AnyRef
  20. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef
  21. def toString(): String

    Definition Classes
    AnyRef → Any
  22. final def wait(): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  23. final def wait(arg0: Long, arg1: Int): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  24. final def wait(arg0: Long): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  25. final def withData(data: PipelineDataset[T], labels: PipelineDataset[DenseVector[Double]]): Pipeline[T, DenseVector[Double]]

    Constructs a pipeline that fits this label estimator to training data and labels, then applies the resultant transformer to the Pipeline input.

    Constructs a pipeline that fits this label estimator to training data and labels, then applies the resultant transformer to the Pipeline input.

    data

    The training data

    labels

    The training labels

    returns

    A pipeline that fits this label estimator and applies the result to inputs.

    Definition Classes
    LabelEstimator
  26. final def withData(data: RDD[T], labels: RDD[DenseVector[Double]]): Pipeline[T, DenseVector[Double]]

    Constructs a pipeline that fits this label estimator to training data and labels, then applies the resultant transformer to the Pipeline input.

    Constructs a pipeline that fits this label estimator to training data and labels, then applies the resultant transformer to the Pipeline input.

    data

    The training data

    labels

    The training labels

    returns

    A pipeline that fits this label estimator and applies the result to inputs.

    Definition Classes
    LabelEstimator
  27. final def withData(data: PipelineDataset[T], labels: RDD[DenseVector[Double]]): Pipeline[T, DenseVector[Double]]

    Constructs a pipeline that fits this label estimator to training data and labels, then applies the resultant transformer to the Pipeline input.

    Constructs a pipeline that fits this label estimator to training data and labels, then applies the resultant transformer to the Pipeline input.

    data

    The training data

    labels

    The training labels

    returns

    A pipeline that fits this label estimator and applies the result to inputs.

    Definition Classes
    LabelEstimator
  28. final def withData(data: RDD[T], labels: PipelineDataset[DenseVector[Double]]): Pipeline[T, DenseVector[Double]]

    Constructs a pipeline that fits this label estimator to training data and labels, then applies the resultant transformer to the Pipeline input.

    Constructs a pipeline that fits this label estimator to training data and labels, then applies the resultant transformer to the Pipeline input.

    data

    The training data

    labels

    The training labels

    returns

    A pipeline that fits this label estimator and applies the result to inputs.

    Definition Classes
    LabelEstimator

Inherited from LabelEstimator[T, DenseVector[Double], DenseVector[Double]]

Inherited from EstimatorOperator

Inherited from Serializable

Inherited from Serializable

Inherited from Operator

Inherited from AnyRef

Inherited from Any

Ungrouped