Class

com.salesforce.op.stages.impl.regression

OpGBTRegressionModel

Related Doc: package regression

Permalink

class OpGBTRegressionModel extends OpPredictionModel[GBTRegressionModel]

Class that takes in a spark GBTRegressionModel and wraps it into an OP model which returns a Prediction feature

Linear Supertypes
OpPredictionModel[GBTRegressionModel], OpPredictorWrapperModel[GBTRegressionModel], OpTransformer2[RealNN, OPVector, Prediction], OpTransformer, OpPipelineStage2[RealNN, OPVector, Prediction], HasIn2, HasIn1, OpPipelineStage[Prediction], OpPipelineStageBase, MLWritable, OpPipelineStageParams, InputParams, SparkWrapperParams[GBTRegressionModel], Model[OpPredictorWrapperModel[GBTRegressionModel]], Transformer, PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. OpGBTRegressionModel
  2. OpPredictionModel
  3. OpPredictorWrapperModel
  4. OpTransformer2
  5. OpTransformer
  6. OpPipelineStage2
  7. HasIn2
  8. HasIn1
  9. OpPipelineStage
  10. OpPipelineStageBase
  11. MLWritable
  12. OpPipelineStageParams
  13. InputParams
  14. SparkWrapperParams
  15. Model
  16. Transformer
  17. PipelineStage
  18. Logging
  19. Params
  20. Serializable
  21. Serializable
  22. Identifiable
  23. AnyRef
  24. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new OpGBTRegressionModel(sparkModel: GBTRegressionModel, uid: String = UID[OpGBTRegressionModel], operationName: String = classOf[GBTRegressor].getSimpleName)(implicit tti1: scala.reflect.api.JavaUniverse.TypeTag[RealNN], tti2: scala.reflect.api.JavaUniverse.TypeTag[OPVector], tto: scala.reflect.api.JavaUniverse.TypeTag[Prediction], ttov: scala.reflect.api.JavaUniverse.TypeTag[Map[String, Double]])

    Permalink

    sparkModel

    model to wrap

    uid

    uid to give stage

    operationName

    unique name of the operation this stage performs

Type Members

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

    Permalink
    Definition Classes
    OpPipelineStage2 → OpPipelineStage → InputParams

  2. type KeyValue = (String) ⇒ Any

    Permalink
    Definition Classes
    OpTransformer

  3. 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. final def checkSerializable: Try[Unit]

    Permalink
    Definition Classes
    OpTransformer2 → OpPipelineStageBase

  8. final def clear(param: Param[_]): OpGBTRegressionModel.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): OpGBTRegressionModel.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. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )

  20. final def get[T](param: Param[T]): Option[T]

    Permalink
    Definition Classes
    Params

  21. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any

  22. final def getDefault[T](param: Param[T]): Option[T]

    Permalink
    Definition Classes
    Params

  23. final def getInputFeature[T <: FeatureType](i: Int): Option[FeatureLike[T]]

    Permalink
    Definition Classes
    InputParams

  24. final def getInputFeatures(): Array[OPFeature]

    Permalink
    Definition Classes
    InputParams

  25. final def getInputSchema(): StructType

    Permalink
    Definition Classes
    OpPipelineStageParams

  26. final def getMetadata(): Metadata

    Permalink
    Definition Classes
    OpPipelineStageParams

  27. final def getOrDefault[T](param: Param[T]): T

    Permalink
    Definition Classes
    Params

  28. def getOutput(): FeatureLike[Prediction]

    Permalink
    Definition Classes
    OpPipelineStage2 → OpPipelineStageBase

  29. final def getOutputFeatureName: String

    Permalink
    Definition Classes
    OpPipelineStage

  30. def getParam(paramName: String): Param[Any]

    Permalink
    Definition Classes
    Params

  31. def getSparkMlStage(): Option[GBTRegressionModel]

    Permalink
    Definition Classes
    SparkWrapperParams

  32. def getStageSavePath(): Option[String]

    Permalink
    Definition Classes
    SparkWrapperParams

  33. final def getTransientFeature(i: Int): Option[TransientFeature]

    Permalink
    Definition Classes
    InputParams

  34. final def getTransientFeatures(): Array[TransientFeature]

    Permalink
    Definition Classes
    InputParams

  35. final def hasDefault[T](param: Param[T]): Boolean

    Permalink
    Definition Classes
    Params

  36. def hasParam(paramName: String): Boolean

    Permalink
    Definition Classes
    Params

  37. def hasParent: Boolean

    Permalink
    Definition Classes
    Model

  38. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any

  39. final def in1: TransientFeature

    Permalink
    Attributes
    protected
    Definition Classes
    HasIn1

  40. final def in2: TransientFeature

    Permalink
    Attributes
    protected
    Definition Classes
    HasIn2

  41. def initializeLogIfNecessary(isInterpreter: Boolean): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  42. final def inputAsArray(in: InputFeatures): Array[OPFeature]

    Permalink
    Definition Classes
    OpPipelineStage2 → InputParams

  43. final def isDefined(param: Param[_]): Boolean

    Permalink
    Definition Classes
    Params

  44. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any

  45. final def isSet(param: Param[_]): Boolean

    Permalink
    Definition Classes
    Params

  46. def isTraceEnabled(): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  47. def log: Logger

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  48. def logDebug(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  49. def logDebug(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  50. def logError(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  51. def logError(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  52. def logInfo(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  53. def logInfo(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  54. def logName: String

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  55. def logTrace(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  56. def logTrace(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  57. def logWarning(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  58. def logWarning(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  59. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef

  60. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef

  61. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef

  62. def onGetMetadata(): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    OpPipelineStageParams

  63. def onSetInput(): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    OpPipelineStageBase

  64. val operationName: String

    Permalink
    Definition Classes
    OpPredictorWrapperModel → OpPipelineStageBase

  65. final def outputAsArray(out: OutputFeatures): Array[OPFeature]

    Permalink
    Definition Classes
    OpPipelineStage → OpPipelineStageBase

  66. def outputFeatureUid: String

    Permalink
    Attributes
    protected[com.salesforce.op]
    Definition Classes
    OpPipelineStage2 → OpPipelineStage

  67. def outputIsResponse: Boolean

    Permalink
    Definition Classes
    OpPipelineStage

  68. lazy val params: Array[Param[_]]

    Permalink
    Definition Classes
    Params

  69. var parent: Estimator[OpPredictorWrapperModel[GBTRegressionModel]]

    Permalink
    Definition Classes
    Model

  70. def predict(features: Vector): Double

    Permalink
    Attributes
    protected
    Definition Classes
    OpPredictionModel

  71. lazy val predictMirror: scala.reflect.api.JavaUniverse.MethodMirror

    Permalink
    Definition Classes
    OpGBTRegressionModel → OpPredictionModel

  72. def save(path: String): Unit

    Permalink
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )

  73. final def set(paramPair: ParamPair[_]): OpGBTRegressionModel.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params

  74. final def set(param: String, value: Any): OpGBTRegressionModel.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params

  75. final def set[T](param: Param[T], value: T): OpGBTRegressionModel.this.type

    Permalink
    Definition Classes
    Params

  76. final def setDefault(paramPairs: ParamPair[_]*): OpGBTRegressionModel.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params

  77. final def setDefault[T](param: Param[T], value: T): OpGBTRegressionModel.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params

  78. final def setInput(features: InputFeatures): OpGBTRegressionModel.this.type

    Permalink
    Definition Classes
    OpPipelineStageBase

  79. final def setInputFeatures[S <: OPFeature](features: Array[S]): OpGBTRegressionModel.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    InputParams

  80. final def setMetadata(m: Metadata): OpGBTRegressionModel.this.type

    Permalink
    Definition Classes
    OpPipelineStageParams

  81. def setOutputFeatureName(name: String): OpGBTRegressionModel.this.type

    Permalink
    Definition Classes
    OpPipelineStage

  82. def setParent(parent: Estimator[OpPredictorWrapperModel[GBTRegressionModel]]): OpPredictorWrapperModel[GBTRegressionModel]

    Permalink
    Definition Classes
    Model

  83. def setSparkMlStage(stage: Option[GBTRegressionModel]): OpGBTRegressionModel.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    SparkWrapperParams

  84. def setStageSavePath(path: String): OpGBTRegressionModel.this.type

    Permalink
    Definition Classes
    SparkWrapperParams

  85. final val sparkInputColParamNames: StringArrayParam

    Permalink
    Definition Classes
    SparkWrapperParams

  86. final val sparkMlStage: SparkStageParam[GBTRegressionModel]

    Permalink
    Definition Classes
    SparkWrapperParams

  87. val sparkModel: GBTRegressionModel

    Permalink
    Definition Classes
    OpPredictorWrapperModel

  88. final val sparkOutputColParamNames: StringArrayParam

    Permalink
    Definition Classes
    SparkWrapperParams

  89. final def stageName: String

    Permalink
    Definition Classes
    OpPipelineStageBase

  90. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef

  91. def toString(): String

    Permalink
    Definition Classes
    Identifiable → AnyRef → Any

  92. def transform(dataset: Dataset[_]): DataFrame

    Permalink
    Definition Classes
    OpTransformer2 → Transformer

  93. def transform(dataset: Dataset[_], paramMap: ParamMap): DataFrame

    Permalink
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" )

  94. def transform(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): DataFrame

    Permalink
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" ) @varargs()

  95. def transformFn: (RealNN, OPVector) ⇒ Prediction

    Permalink

    Function used to convert input to output

    Function used to convert input to output

    Definition Classes
    OpPredictionModel → OpTransformer2

  96. lazy val transformKeyValue: (KeyValue) ⇒ Any

    Permalink
    Definition Classes
    OpTransformer2 → OpTransformer

  97. def transformMap: (Map[String, Any]) ⇒ Any

    Permalink
    Definition Classes
    OpTransformer

  98. def transformRow: (Row) ⇒ Any

    Permalink
    Definition Classes
    OpTransformer

  99. final def transformSchema(schema: StructType): StructType

    Permalink
    Definition Classes
    OpPipelineStageBase

  100. def transformSchema(schema: StructType, logging: Boolean): StructType

    Permalink
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()

  101. implicit val tti1: scala.reflect.api.JavaUniverse.TypeTag[RealNN]

    Permalink
    Definition Classes
    OpPredictorWrapperModel → OpTransformer2

  102. implicit val tti2: scala.reflect.api.JavaUniverse.TypeTag[OPVector]

    Permalink
    Definition Classes
    OpPredictorWrapperModel → OpTransformer2

  103. implicit val tto: scala.reflect.api.JavaUniverse.TypeTag[Prediction]

    Permalink
    Definition Classes
    OpPredictorWrapperModel → OpPipelineStage2

  104. implicit val ttov: scala.reflect.api.JavaUniverse.TypeTag[Map[String, Double]]

    Permalink
    Definition Classes
    OpPredictorWrapperModel → OpPipelineStage2

  105. val uid: String

    Permalink
    Definition Classes
    OpPredictorWrapperModel → Identifiable

  106. final def wait(): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  107. final def wait(arg0: Long, arg1: Int): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  108. final def wait(arg0: Long): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  109. final def write: MLWriter

    Permalink
    Definition Classes
    OpPipelineStageBase → MLWritable

Inherited from OpPredictionModel[GBTRegressionModel]

Inherited from OpPredictorWrapperModel[GBTRegressionModel]

Inherited from OpTransformer2[RealNN, OPVector, Prediction]

Inherited from OpTransformer

Inherited from OpPipelineStage2[RealNN, OPVector, 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 SparkWrapperParams[GBTRegressionModel]

Inherited from Model[OpPredictorWrapperModel[GBTRegressionModel]]

Inherited from Transformer

Inherited from PipelineStage

Inherited from Logging

Inherited from Params

Inherited from Serializable

Inherited from Serializable

Inherited from Identifiable

Inherited from AnyRef

Inherited from Any

Ungrouped