Class

com.salesforce.op.stages.impl.regression

OpXGBoostRegressionModel

Related Doc: package regression

Permalink

class OpXGBoostRegressionModel extends OpPredictionModel[XGBoostRegressionModel]

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

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

Instance Constructors

  1. new OpXGBoostRegressionModel(sparkModel: XGBoostRegressionModel, uid: String = UID[OpXGBoostRegressionModel], operationName: String = ...)(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[_]): OpXGBoostRegressionModel.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): OpXGBoostRegressionModel.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. def getInputColParamNames(): Array[String]

    Permalink
    Definition Classes
    SparkWrapperParams

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

    Permalink
    Definition Classes
    InputParams

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

    Permalink
    Definition Classes
    InputParams

  26. final def getInputSchema(): StructType

    Permalink
    Definition Classes
    OpPipelineStageParams

  27. final def getMetadata(): Metadata

    Permalink
    Definition Classes
    OpPipelineStageParams

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

    Permalink
    Definition Classes
    Params

  29. def getOutput(): FeatureLike[Prediction]

    Permalink
    Definition Classes
    HasOut → OpPipelineStageBase

  30. def getOutputColParamNames(): Array[String]

    Permalink
    Definition Classes
    SparkWrapperParams

  31. final def getOutputFeatureName: String

    Permalink
    Definition Classes
    OpPipelineStage

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

    Permalink
    Definition Classes
    Params

  33. def getSparkMlStage(): Option[XGBoostRegressionModel]

    Permalink
    Definition Classes
    SparkWrapperParams

  34. def getStageSavePath(): Option[String]

    Permalink
    Definition Classes
    SparkWrapperParams

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

    Permalink
    Definition Classes
    InputParams

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

    Permalink
    Definition Classes
    InputParams

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

    Permalink
    Definition Classes
    Params

  38. def hasParam(paramName: String): Boolean

    Permalink
    Definition Classes
    Params

  39. def hasParent: Boolean

    Permalink
    Definition Classes
    Model

  40. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any

  41. final def in1: TransientFeature

    Permalink
    Attributes
    protected
    Definition Classes
    HasIn1

  42. final def in2: TransientFeature

    Permalink
    Attributes
    protected
    Definition Classes
    HasIn2

  43. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  44. def initializeLogIfNecessary(isInterpreter: Boolean): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Definition Classes
    OpPipelineStage2 → InputParams

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

    Permalink
    Definition Classes
    Params

  47. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any

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

    Permalink
    Definition Classes
    Params

  49. def isTraceEnabled(): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  50. def log: Logger

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

  57. def logName: String

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Attributes
    protected
    Definition Classes
    Logging

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

    Permalink
    Definition Classes
    AnyRef

  63. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef

  64. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef

  65. def onGetMetadata(): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    OpPipelineStageParams

  66. def onSetInput(): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    InputParams

  67. val operationName: String

    Permalink
    Definition Classes
    OpPredictorWrapperModel → OpPipelineStageBase

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

    Permalink
    Definition Classes
    OpPipelineStage → OpPipelineStageBase

  69. def outputFeatureUid: String

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

  70. def outputIsResponse: Boolean

    Permalink
    Definition Classes
    OpPipelineStage

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

    Permalink
    Definition Classes
    Params

  72. var parent: Estimator[OpPredictorWrapperModel[XGBoostRegressionModel]]

    Permalink
    Definition Classes
    Model

  73. lazy val predict: (Vector) ⇒ Double

    Permalink

    Predict label for the given features

    Predict label for the given features

    Attributes
    protected
    Definition Classes
    OpPredictionModel

  74. def save(path: String): Unit

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

  75. final def set(paramPair: ParamPair[_]): OpXGBoostRegressionModel.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params

  76. final def set(param: String, value: Any): OpXGBoostRegressionModel.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params

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

    Permalink
    Definition Classes
    Params

  78. final def setDefault(paramPairs: ParamPair[_]*): OpXGBoostRegressionModel.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params

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

    Permalink
    Attributes
    protected
    Definition Classes
    Params

  80. final def setInput(features: InputFeatures): OpXGBoostRegressionModel.this.type

    Permalink
    Definition Classes
    OpPipelineStageBase

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

    Permalink
    Attributes
    protected
    Definition Classes
    InputParams

  82. final def setMetadata(m: Metadata): OpXGBoostRegressionModel.this.type

    Permalink
    Definition Classes
    OpPipelineStageParams

  83. def setOutputFeatureName(name: String): OpXGBoostRegressionModel.this.type

    Permalink
    Definition Classes
    OpPipelineStage

  84. def setParent(parent: Estimator[OpPredictorWrapperModel[XGBoostRegressionModel]]): OpPredictorWrapperModel[XGBoostRegressionModel]

    Permalink
    Definition Classes
    Model

  85. def setSparkMlStage(stage: Option[XGBoostRegressionModel]): OpXGBoostRegressionModel.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    SparkWrapperParams

  86. def setStageSavePath(path: String): OpXGBoostRegressionModel.this.type

    Permalink
    Definition Classes
    SparkWrapperParams

  87. final val sparkInputColParamNames: StringArrayParam

    Permalink
    Definition Classes
    SparkWrapperParams

  88. final val sparkMlStage: SparkStageParam[XGBoostRegressionModel]

    Permalink
    Definition Classes
    SparkWrapperParams

  89. val sparkModel: XGBoostRegressionModel

    Permalink
    Definition Classes
    OpPredictorWrapperModel

  90. final val sparkOutputColParamNames: StringArrayParam

    Permalink
    Definition Classes
    SparkWrapperParams

  91. final def stageName: String

    Permalink
    Definition Classes
    OpPipelineStageBase

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

    Permalink
    Definition Classes
    AnyRef

  93. def toString(): String

    Permalink
    Definition Classes
    Identifiable → AnyRef → Any

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

    Permalink
    Definition Classes
    OpTransformer2 → Transformer

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

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

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

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

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

    Permalink

    Function used to convert input to output

    Function used to convert input to output

    Definition Classes
    OpPredictionModel → OpTransformer2

  98. lazy val transformKeyValue: (KeyValue) ⇒ Any

    Permalink
    Definition Classes
    OpTransformer2 → OpTransformer

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

    Permalink
    Definition Classes
    OpTransformer

  100. def transformRow: (Row) ⇒ Any

    Permalink
    Definition Classes
    OpTransformer

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

    Permalink
    Definition Classes
    OpPipelineStageBase

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

    Permalink
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()

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

    Permalink
    Definition Classes
    OpPredictorWrapperModel → OpTransformer2

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

    Permalink
    Definition Classes
    OpPredictorWrapperModel → OpTransformer2

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

    Permalink
    Definition Classes
    OpPredictorWrapperModel → HasOut

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

    Permalink
    Definition Classes
    OpPredictorWrapperModel → HasOut

  107. val uid: String

    Permalink
    Definition Classes
    OpPredictorWrapperModel → Identifiable

  108. final def wait(): Unit

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

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

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

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

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

  111. final def write: MLWriter

    Permalink
    Definition Classes
    OpPipelineStageBase → MLWritable

Inherited from OpPredictionModel[XGBoostRegressionModel]

Inherited from OpPredictorWrapperModel[XGBoostRegressionModel]

Inherited from OpTransformer2[RealNN, OPVector, Prediction]

Inherited from OpTransformer

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 SparkWrapperParams[XGBoostRegressionModel]

Inherited from Model[OpPredictorWrapperModel[XGBoostRegressionModel]]

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