Package org.apache.spark.mllib.regression

Class RidgeRegressionWithSGD

Object
org.apache.spark.mllib.regression.GeneralizedLinearAlgorithm<RidgeRegressionModel>
org.apache.spark.mllib.regression.RidgeRegressionWithSGD
All Implemented Interfaces:
Serializable, org.apache.spark.internal.Logging
public class RidgeRegressionWithSGD extends GeneralizedLinearAlgorithm<RidgeRegressionModel> implements Serializable
Train a regression model with L2-regularization using Stochastic Gradient Descent. This solves the l2-regularized least squares regression formulation f(weights) = 1/2n ||A weights-y||^2^ + regParam/2 ||weights||^2^ Here the data matrix has n rows, and the input RDD holds the set of rows of A, each with its corresponding right hand side label y. See also the documentation for the precise formulation.
See Also:

  • Nested Class Summary

    Nested classes/interfaces inherited from interface org.apache.spark.internal.Logging

    org.apache.spark.internal.Logging.LogStringContext, org.apache.spark.internal.Logging.SparkShellLoggingFilter

  • Method Summary

    Modifier and Type
    Method
    Description
    GradientDescent
    optimizer()
    The optimizer to solve the problem.

    Methods inherited from class org.apache.spark.mllib.regression.GeneralizedLinearAlgorithm

    getNumFeatures, isAddIntercept, run, run, setIntercept, setValidateData

    Methods inherited from class java.lang.Object

    equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

    Methods inherited from interface org.apache.spark.internal.Logging

    initializeForcefully, initializeLogIfNecessary, initializeLogIfNecessary, initializeLogIfNecessary$default$2, isTraceEnabled, log, logBasedOnLevel, logDebug, logDebug, logDebug, logDebug, logError, logError, logError, logError, logInfo, logInfo, logInfo, logInfo, logName, LogStringContext, logTrace, logTrace, logTrace, logTrace, logWarning, logWarning, logWarning, logWarning, org$apache$spark$internal$Logging$$log_, org$apache$spark$internal$Logging$$log__$eq, withLogContext