Class

com.salesforce.op.stages.impl.regression

OpGeneralizedLinearRegression

Related Doc: package regression

Permalink

class OpGeneralizedLinearRegression extends OpPredictorWrapper[GeneralizedLinearRegression, GeneralizedLinearRegressionModel] with OpGeneralizedLinearRegressionParams

Wrapper for spark Generalized Regression org.apache.spark.ml.regression.GeneralizedLinearRegression

Linear Supertypes
OpGeneralizedLinearRegressionParams, GeneralizedLinearRegressionBase, HasSolver, HasWeightCol, HasRegParam, HasTol, HasMaxIter, HasFitIntercept, PredictorParams, HasPredictionCol, HasFeaturesCol, HasLabelCol, OpPredictorWrapper[GeneralizedLinearRegression, GeneralizedLinearRegressionModel], SparkWrapperParams[GeneralizedLinearRegression], OpPipelineStage2[RealNN, OPVector, Prediction], HasOut[Prediction], HasIn2, HasIn1, OpPipelineStage[Prediction], OpPipelineStageBase, MLWritable, OpPipelineStageParams, InputParams, Estimator[OpPredictorWrapperModel[GeneralizedLinearRegressionModel]], PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. OpGeneralizedLinearRegression
  2. OpGeneralizedLinearRegressionParams
  3. GeneralizedLinearRegressionBase
  4. HasSolver
  5. HasWeightCol
  6. HasRegParam
  7. HasTol
  8. HasMaxIter
  9. HasFitIntercept
  10. PredictorParams
  11. HasPredictionCol
  12. HasFeaturesCol
  13. HasLabelCol
  14. OpPredictorWrapper
  15. SparkWrapperParams
  16. OpPipelineStage2
  17. HasOut
  18. HasIn2
  19. HasIn1
  20. OpPipelineStage
  21. OpPipelineStageBase
  22. MLWritable
  23. OpPipelineStageParams
  24. InputParams
  25. Estimator
  26. PipelineStage
  27. Logging
  28. Params
  29. Serializable
  30. Serializable
  31. Identifiable
  32. AnyRef
  33. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new OpGeneralizedLinearRegression(uid: String = UID[OpGeneralizedLinearRegression])

    Permalink

    uid

    stage uid

Type Members

  1. final type InputFeatures = (FeatureLike[RealNN], FeatureLike[OPVector])

    Permalink
    Definition Classes
    OpPipelineStage2 → OpPipelineStage → InputParams
  2. final type OutputFeatures = FeatureLike[Prediction]

    Permalink
    Definition Classes
    OpPipelineStage → OpPipelineStageBase

Value Members

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

    Permalink
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  3. final def $[T](param: Param[T]): T

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  4. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  5. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any
  6. final def checkInputLength(features: Array[_]): Boolean

    Permalink
    Definition Classes
    OpPipelineStage2 → InputParams
  7. def checkSerializable: Try[Unit]

    Permalink
    Definition Classes
    OpPipelineStageBase
  8. final def clear(param: Param[_]): OpGeneralizedLinearRegression.this.type

    Permalink
    Definition Classes
    Params
  9. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  10. final def copy(extra: ParamMap): OpGeneralizedLinearRegression.this.type

    Permalink
    Definition Classes
    OpPipelineStageBase → Params
  11. def copyValues[T <: Params](to: T, extra: ParamMap): T

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  12. final def defaultCopy[T <: Params](extra: ParamMap): T

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  13. final def eq(arg0: AnyRef): Boolean

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

    Permalink
    Definition Classes
    AnyRef → Any
  15. def explainParam(param: Param[_]): String

    Permalink
    Definition Classes
    Params
  16. def explainParams(): String

    Permalink
    Definition Classes
    Params
  17. final def extractParamMap(): ParamMap

    Permalink
    Definition Classes
    Params
  18. final def extractParamMap(extra: ParamMap): ParamMap

    Permalink
    Definition Classes
    Params
  19. final val family: Param[String]

    Permalink
    Definition Classes
    GeneralizedLinearRegressionBase
    Annotations
    @Since( "2.0.0" )
  20. final val featuresCol: Param[String]

    Permalink
    Definition Classes
    HasFeaturesCol
  21. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  22. def fit(dataset: Dataset[_]): OpPredictorWrapperModel[GeneralizedLinearRegressionModel]

    Permalink

    Function that fits the binary model

    Function that fits the binary model

    Definition Classes
    OpPredictorWrapper → Estimator
  23. def fit(dataset: Dataset[_], paramMaps: Array[ParamMap]): Seq[OpPredictorWrapperModel[GeneralizedLinearRegressionModel]]

    Permalink
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  24. def fit(dataset: Dataset[_], paramMap: ParamMap): OpPredictorWrapperModel[GeneralizedLinearRegressionModel]

    Permalink
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  25. def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): OpPredictorWrapperModel[GeneralizedLinearRegressionModel]

    Permalink
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" ) @varargs()
  26. final val fitIntercept: BooleanParam

    Permalink
    Definition Classes
    HasFitIntercept
  27. final def get[T](param: Param[T]): Option[T]

    Permalink
    Definition Classes
    Params
  28. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any
  29. final def getDefault[T](param: Param[T]): Option[T]

    Permalink
    Definition Classes
    Params
  30. def getFamily: String

    Permalink
    Definition Classes
    GeneralizedLinearRegressionBase
    Annotations
    @Since( "2.0.0" )
  31. final def getFeaturesCol: String

    Permalink
    Definition Classes
    HasFeaturesCol
  32. final def getFitIntercept: Boolean

    Permalink
    Definition Classes
    HasFitIntercept
  33. def getInputColParamNames(): Array[String]

    Permalink
    Definition Classes
    SparkWrapperParams
  34. final def getInputFeature[T <: FeatureType](i: Int): Option[FeatureLike[T]]

    Permalink
    Definition Classes
    InputParams
  35. final def getInputFeatures(): Array[OPFeature]

    Permalink
    Definition Classes
    InputParams
  36. final def getInputSchema(): StructType

    Permalink
    Definition Classes
    OpPipelineStageParams
  37. final def getLabelCol: String

    Permalink
    Definition Classes
    HasLabelCol
  38. def getLink: String

    Permalink
    Definition Classes
    GeneralizedLinearRegressionBase
    Annotations
    @Since( "2.0.0" )
  39. def getLinkPower: Double

    Permalink
    Definition Classes
    GeneralizedLinearRegressionBase
    Annotations
    @Since( "2.2.0" )
  40. def getLinkPredictionCol: String

    Permalink
    Definition Classes
    GeneralizedLinearRegressionBase
    Annotations
    @Since( "2.0.0" )
  41. final def getMaxIter: Int

    Permalink
    Definition Classes
    HasMaxIter
  42. final def getMetadata(): Metadata

    Permalink
    Definition Classes
    OpPipelineStageParams
  43. def getOffsetCol: String

    Permalink
    Definition Classes
    GeneralizedLinearRegressionBase
    Annotations
    @Since( "2.3.0" )
  44. final def getOrDefault[T](param: Param[T]): T

    Permalink
    Definition Classes
    Params
  45. def getOutput(): FeatureLike[Prediction]

    Permalink
    Definition Classes
    HasOut → OpPipelineStageBase
  46. def getOutputColParamNames(): Array[String]

    Permalink
    Definition Classes
    SparkWrapperParams
  47. final def getOutputFeatureName: String

    Permalink
    Definition Classes
    OpPipelineStage
  48. def getParam(paramName: String): Param[Any]

    Permalink
    Definition Classes
    Params
  49. final def getPredictionCol: String

    Permalink
    Definition Classes
    HasPredictionCol
  50. final def getRegParam: Double

    Permalink
    Definition Classes
    HasRegParam
  51. final def getSolver: String

    Permalink
    Definition Classes
    HasSolver
  52. def getSparkMlStage(): Option[GeneralizedLinearRegression]

    Permalink
    Definition Classes
    SparkWrapperParams
  53. def getStageSavePath(): Option[String]

    Permalink
    Definition Classes
    SparkWrapperParams
  54. final def getTol: Double

    Permalink
    Definition Classes
    HasTol
  55. final def getTransientFeature(i: Int): Option[TransientFeature]

    Permalink
    Definition Classes
    InputParams
  56. final def getTransientFeatures(): Array[TransientFeature]

    Permalink
    Definition Classes
    InputParams
  57. def getVariancePower: Double

    Permalink
    Definition Classes
    GeneralizedLinearRegressionBase
    Annotations
    @Since( "2.2.0" )
  58. final def getWeightCol: String

    Permalink
    Definition Classes
    HasWeightCol
  59. final def hasDefault[T](param: Param[T]): Boolean

    Permalink
    Definition Classes
    Params
  60. def hasParam(paramName: String): Boolean

    Permalink
    Definition Classes
    Params
  61. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  62. final def in1: TransientFeature

    Permalink
    Attributes
    protected
    Definition Classes
    HasIn1
  63. final def in2: TransientFeature

    Permalink
    Attributes
    protected
    Definition Classes
    HasIn2
  64. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  65. def initializeLogIfNecessary(isInterpreter: Boolean): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  66. final def inputAsArray(in: InputFeatures): Array[OPFeature]

    Permalink
    Definition Classes
    OpPipelineStage2 → InputParams
  67. val inputParam1Name: String

    Permalink
    Definition Classes
    OpPredictorWrapper
  68. val inputParam2Name: String

    Permalink
    Definition Classes
    OpPredictorWrapper
  69. final def isDefined(param: Param[_]): Boolean

    Permalink
    Definition Classes
    Params
  70. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any
  71. final def isSet(param: Param[_]): Boolean

    Permalink
    Definition Classes
    Params
  72. def isTraceEnabled(): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  73. final val labelCol: Param[String]

    Permalink
    Definition Classes
    HasLabelCol
  74. final val link: Param[String]

    Permalink
    Definition Classes
    GeneralizedLinearRegressionBase
    Annotations
    @Since( "2.0.0" )
  75. final val linkPower: DoubleParam

    Permalink
    Definition Classes
    GeneralizedLinearRegressionBase
    Annotations
    @Since( "2.2.0" )
  76. final val linkPredictionCol: Param[String]

    Permalink
    Definition Classes
    GeneralizedLinearRegressionBase
    Annotations
    @Since( "2.0.0" )
  77. def log: Logger

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  78. def logDebug(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  79. def logDebug(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  80. def logError(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  81. def logError(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  82. def logInfo(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  83. def logInfo(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  84. def logName: String

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  85. def logTrace(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  86. def logTrace(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  87. def logWarning(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  88. def logWarning(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  89. final val maxIter: IntParam

    Permalink
    Definition Classes
    HasMaxIter
  90. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  91. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef
  92. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef
  93. final val offsetCol: Param[String]

    Permalink
    Definition Classes
    GeneralizedLinearRegressionBase
    Annotations
    @Since( "2.3.0" )
  94. def onGetMetadata(): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    OpPipelineStageParams
  95. def onSetInput(): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    OpGeneralizedLinearRegression → InputParams
  96. val operationName: String

    Permalink
    Definition Classes
    OpPredictorWrapper → OpPipelineStageBase
  97. final def outputAsArray(out: OutputFeatures): Array[OPFeature]

    Permalink
    Definition Classes
    OpPipelineStage → OpPipelineStageBase
  98. def outputFeatureUid: String

    Permalink
    Attributes
    protected[com.salesforce.op]
    Definition Classes
    OpPipelineStage2 → OpPipelineStage
  99. def outputIsResponse: Boolean

    Permalink
    Definition Classes
    OpPipelineStage
  100. val outputParamName: String

    Permalink
    Definition Classes
    OpPredictorWrapper
  101. lazy val params: Array[Param[_]]

    Permalink
    Definition Classes
    Params
  102. final val predictionCol: Param[String]

    Permalink
    Definition Classes
    HasPredictionCol
  103. val predictor: GeneralizedLinearRegression

    Permalink

    the predictor to wrap

    the predictor to wrap

    Definition Classes
    OpPredictorWrapper
  104. final val regParam: DoubleParam

    Permalink
    Definition Classes
    HasRegParam
  105. def save(path: String): Unit

    Permalink
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )
  106. final def set(paramPair: ParamPair[_]): OpGeneralizedLinearRegression.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  107. final def set(param: String, value: Any): OpGeneralizedLinearRegression.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  108. final def set[T](param: Param[T], value: T): OpGeneralizedLinearRegression.this.type

    Permalink
    Definition Classes
    Params
  109. final def setDefault(paramPairs: ParamPair[_]*): OpGeneralizedLinearRegression.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  110. final def setDefault[T](param: Param[T], value: T): OpGeneralizedLinearRegression.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  111. def setFamily(value: String): OpGeneralizedLinearRegression.this.type

    Permalink

    Sets the value of param family.

    Sets the value of param family. Default is "gaussian".

  112. def setFitIntercept(value: Boolean): OpGeneralizedLinearRegression.this.type

    Permalink

    Sets if we should fit the intercept.

    Sets if we should fit the intercept. Default is true.

  113. final def setInput(features: InputFeatures): OpGeneralizedLinearRegression.this.type

    Permalink
    Definition Classes
    OpPipelineStageBase
  114. final def setInputFeatures[S <: OPFeature](features: Array[S]): OpGeneralizedLinearRegression.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    InputParams
  115. def setLink(value: String): OpGeneralizedLinearRegression.this.type

    Permalink

    Sets the value of param link.

    Sets the value of param link. Used only when family is not "tweedie".

  116. def setLinkPower(value: Double): OpGeneralizedLinearRegression.this.type

    Permalink

    Sets the value of param linkPower.

    Sets the value of param linkPower. Used only when family is "tweedie".

  117. def setLinkPredictionCol(value: String): OpGeneralizedLinearRegression.this.type

    Permalink

    Sets the link prediction (linear predictor) column name.

  118. def setMaxIter(value: Int): OpGeneralizedLinearRegression.this.type

    Permalink

    Sets the maximum number of iterations (applicable for solver "irls").

    Sets the maximum number of iterations (applicable for solver "irls"). Default is 25.

  119. final def setMetadata(m: Metadata): OpGeneralizedLinearRegression.this.type

    Permalink
    Definition Classes
    OpPipelineStageParams
  120. def setOutputFeatureName(name: String): OpGeneralizedLinearRegression.this.type

    Permalink
    Definition Classes
    OpPipelineStage
  121. def setRegParam(value: Double): OpGeneralizedLinearRegression.this.type

    Permalink

    Sets the regularization parameter for L2 regularization.

    Sets the regularization parameter for L2 regularization. The regularization term is

    $$ 0.5 * regParam * L2norm(coefficients)^2 $$
    Default is 0.0.

  122. def setSolver(value: String): OpGeneralizedLinearRegression.this.type

    Permalink

    Sets the solver algorithm used for optimization.

    Sets the solver algorithm used for optimization. Currently only supports "irls" which is also the default solver.

  123. def setSparkMlStage(stage: Option[GeneralizedLinearRegression]): OpGeneralizedLinearRegression.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    SparkWrapperParams
  124. def setStageSavePath(path: String): OpGeneralizedLinearRegression.this.type

    Permalink
    Definition Classes
    SparkWrapperParams
  125. def setTol(value: Double): OpGeneralizedLinearRegression.this.type

    Permalink

    Sets the convergence tolerance of iterations.

    Sets the convergence tolerance of iterations. Smaller value will lead to higher accuracy with the cost of more iterations. Default is 1E-6.

  126. def setVariancePower(value: Double): OpGeneralizedLinearRegression.this.type

    Permalink

    Sets the value of param variancePower.

    Sets the value of param variancePower. Used only when family is "tweedie". Default is 0.0, which corresponds to the "gaussian" family.

  127. def setWeightCol(value: String): OpGeneralizedLinearRegression.this.type

    Permalink

    Sets the value of param weightCol.

    Sets the value of param weightCol. If this is not set or empty, we treat all instance weights as 1.0. Default is not set, so all instances have weight one. In the Binomial family, weights correspond to number of trials and should be integer. Non-integer weights are rounded to integer in AIC calculation.

  128. final val solver: Param[String]

    Permalink
    Definition Classes
    GeneralizedLinearRegressionBase → HasSolver
    Annotations
    @Since( "2.0.0" )
  129. final val sparkInputColParamNames: StringArrayParam

    Permalink
    Definition Classes
    SparkWrapperParams
  130. final val sparkMlStage: SparkStageParam[GeneralizedLinearRegression]

    Permalink
    Definition Classes
    SparkWrapperParams
  131. final val sparkOutputColParamNames: StringArrayParam

    Permalink
    Definition Classes
    SparkWrapperParams
  132. final def stageName: String

    Permalink
    Definition Classes
    OpPipelineStageBase
  133. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef
  134. def toString(): String

    Permalink
    Definition Classes
    Identifiable → AnyRef → Any
  135. final val tol: DoubleParam

    Permalink
    Definition Classes
    HasTol
  136. final def transformSchema(schema: StructType): StructType

    Permalink
    Definition Classes
    OpPipelineStageBase
  137. def transformSchema(schema: StructType, logging: Boolean): StructType

    Permalink
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  138. implicit val tti1: scala.reflect.api.JavaUniverse.TypeTag[RealNN]

    Permalink
    Definition Classes
    OpPredictorWrapper
  139. implicit val tti2: scala.reflect.api.JavaUniverse.TypeTag[OPVector]

    Permalink
    Definition Classes
    OpPredictorWrapper
  140. implicit val tto: scala.reflect.api.JavaUniverse.TypeTag[Prediction]

    Permalink
    Definition Classes
    OpPredictorWrapper → HasOut
  141. implicit val ttov: scala.reflect.api.JavaUniverse.TypeTag[Map[String, Double]]

    Permalink
    Definition Classes
    OpPredictorWrapper → HasOut
  142. val uid: String

    Permalink

    stage uid

    stage uid

    Definition Classes
    OpPredictorWrapper → Identifiable
  143. def validateAndTransformSchema(schema: StructType, fitting: Boolean, featuresDataType: DataType): StructType

    Permalink
    Definition Classes
    GeneralizedLinearRegressionBase → PredictorParams
    Annotations
    @Since( "2.0.0" )
  144. final val variancePower: DoubleParam

    Permalink
    Definition Classes
    GeneralizedLinearRegressionBase
    Annotations
    @Since( "2.2.0" )
  145. final def wait(): Unit

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

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

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  148. final val weightCol: Param[String]

    Permalink
    Definition Classes
    HasWeightCol
  149. final def write: MLWriter

    Permalink
    Definition Classes
    OpPipelineStageBase → MLWritable

Inherited from GeneralizedLinearRegressionBase

Inherited from HasSolver

Inherited from HasWeightCol

Inherited from HasRegParam

Inherited from HasTol

Inherited from HasMaxIter

Inherited from HasFitIntercept

Inherited from PredictorParams

Inherited from HasPredictionCol

Inherited from HasFeaturesCol

Inherited from HasLabelCol

Inherited from OpPredictorWrapper[GeneralizedLinearRegression, GeneralizedLinearRegressionModel]

Inherited from SparkWrapperParams[GeneralizedLinearRegression]

Inherited from OpPipelineStage2[RealNN, OPVector, Prediction]

Inherited from HasOut[Prediction]

Inherited from HasIn2

Inherited from HasIn1

Inherited from OpPipelineStage[Prediction]

Inherited from OpPipelineStageBase

Inherited from MLWritable

Inherited from OpPipelineStageParams

Inherited from InputParams

Inherited from Estimator[OpPredictorWrapperModel[GeneralizedLinearRegressionModel]]

Inherited from PipelineStage

Inherited from Logging

Inherited from Params

Inherited from Serializable

Inherited from Serializable

Inherited from Identifiable

Inherited from AnyRef

Inherited from Any

setParam

Ungrouped