Packages

class XGBoostRegressionModel extends PredictionModel[Vector, XGBoostRegressionModel] with RankerRegressorBaseModel[XGBoostRegressionModel]

Linear Supertypes
RankerRegressorBaseModel[XGBoostRegressionModel], XGBPredictorParams[XGBoostRegressionModel], XGBoostModel[XGBoostRegressionModel], ParamUtils[XGBoostRegressionModel], SparkParams[XGBoostRegressionModel], SchemaValidationTrait, RabitParams, NonXGBoostParams, HasContribPredictionCol, HasLeafPredictionCol, HasWeightCol, HasBaseMarginCol, HasFeaturesCols, XGBoostParams[XGBoostRegressionModel], DartBoosterParams, GeneralParams, LearningTaskParams, TreeBoosterParams, MLWritable, PredictionModel[Vector, XGBoostRegressionModel], PredictorParams, HasPredictionCol, HasFeaturesCol, HasLabelCol, Model[XGBoostRegressionModel], Transformer, PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. XGBoostRegressionModel
  2. RankerRegressorBaseModel
  3. XGBPredictorParams
  4. XGBoostModel
  5. ParamUtils
  6. SparkParams
  7. SchemaValidationTrait
  8. RabitParams
  9. NonXGBoostParams
  10. HasContribPredictionCol
  11. HasLeafPredictionCol
  12. HasWeightCol
  13. HasBaseMarginCol
  14. HasFeaturesCols
  15. XGBoostParams
  16. DartBoosterParams
  17. GeneralParams
  18. LearningTaskParams
  19. TreeBoosterParams
  20. MLWritable
  21. PredictionModel
  22. PredictorParams
  23. HasPredictionCol
  24. HasFeaturesCol
  25. HasLabelCol
  26. Model
  27. Transformer
  28. PipelineStage
  29. Logging
  30. Params
  31. Serializable
  32. Serializable
  33. Identifiable
  34. AnyRef
  35. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new XGBoostRegressionModel(uid: String)

Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. final def $[T](param: Param[T]): T
    Attributes
    protected
    Definition Classes
    Params
  4. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  5. val TMP_TRANSFORMED_COL: String
    Attributes
    protected
    Definition Classes
    XGBoostModel
  6. def addNonXGBoostParam(ps: Param[_]*): Unit
    Attributes
    protected
    Definition Classes
    NonXGBoostParams
  7. final val aftLossDistribution: Param[String]
    Definition Classes
    LearningTaskParams
  8. final val alpha: DoubleParam
    Definition Classes
    TreeBoosterParams
  9. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  10. final val baseMarginCol: Param[String]

    Param for initial prediction (aka base margin) column name.

    Param for initial prediction (aka base margin) column name.

    Definition Classes
    HasBaseMarginCol
  11. final val baseScore: DoubleParam
    Definition Classes
    LearningTaskParams
  12. final val booster: Param[String]
    Definition Classes
    GeneralParams
  13. final val cacheHostRatio: FloatParam
    Definition Classes
    SparkParams
  14. final def clear(param: Param[_]): XGBoostRegressionModel.this.type
    Definition Classes
    Params
  15. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native() @IntrinsicCandidate()
  16. final val colsampleBylevel: DoubleParam
    Definition Classes
    TreeBoosterParams
  17. final val colsampleBynode: DoubleParam
    Definition Classes
    TreeBoosterParams
  18. final val colsampleBytree: DoubleParam
    Definition Classes
    TreeBoosterParams
  19. final val contribPredictionCol: Param[String]

    Param for contribution prediction column name.

    Param for contribution prediction column name.

    Definition Classes
    HasContribPredictionCol
  20. def copy(extra: ParamMap): XGBoostRegressionModel
    Definition Classes
    XGBoostRegressionModel → XGBoostModel → Model → Transformer → PipelineStage → Params
  21. def copyValues[T <: Params](to: T, extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  22. final val customEval: CustomEvalParam
    Definition Classes
    SparkParams
  23. final val customObj: CustomObjParam
    Definition Classes
    SparkParams
  24. final def defaultCopy[T <: Params](extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  25. final val device: Param[String]
    Definition Classes
    GeneralParams
  26. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  27. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  28. final val eta: DoubleParam
    Definition Classes
    TreeBoosterParams
  29. final val evalMetric: Param[String]
    Definition Classes
    LearningTaskParams
  30. final val expectileAlpha: Param[String]
    Definition Classes
    LearningTaskParams
  31. def explainParam(param: Param[_]): String
    Definition Classes
    Params
  32. def explainParams(): String
    Definition Classes
    Params
  33. final def extractParamMap(): ParamMap
    Definition Classes
    Params
  34. final def extractParamMap(extra: ParamMap): ParamMap
    Definition Classes
    Params
  35. def featureIsArrayType(schema: StructType): Boolean
    Attributes
    protected[spark]
    Definition Classes
    SparkParams
  36. final val featureNames: StringArrayParam

    Feature's name, it will be set to DMatrix and Booster, and in the final native json model.

    Feature's name, it will be set to DMatrix and Booster, and in the final native json model. In native code, the parameter name is feature_name.

    Definition Classes
    SparkParams
  37. final val featureTypes: StringArrayParam

    Feature types, q is numeric and c is categorical.

    Feature types, q is numeric and c is categorical. In native code, the parameter name is feature_type

    Definition Classes
    SparkParams
  38. final val featuresCol: Param[String]
    Definition Classes
    HasFeaturesCol
  39. final val featuresCols: StringArrayParam

    Param for the names of feature columns.

    Param for the names of feature columns.

    Definition Classes
    HasFeaturesCols
  40. def featuresDataType: DataType
    Attributes
    protected
    Definition Classes
    PredictionModel
  41. final val forceRepartition: BooleanParam
    Definition Classes
    SparkParams
  42. final val gamma: DoubleParam
    Definition Classes
    TreeBoosterParams
  43. final def get[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  44. final def getAftLossDistribution: String
    Definition Classes
    LearningTaskParams
  45. final def getAlpha: Double
    Definition Classes
    TreeBoosterParams
  46. final def getBaseMarginCol: String

    Definition Classes
    HasBaseMarginCol
  47. final def getBaseScore: Double
    Definition Classes
    LearningTaskParams
  48. final def getBooster: String
    Definition Classes
    GeneralParams
  49. final def getCacheHostRatio: Float
    Definition Classes
    SparkParams
  50. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native() @IntrinsicCandidate()
  51. final def getColsampleBylevel: Double
    Definition Classes
    TreeBoosterParams
  52. final def getColsampleBynode: Double
    Definition Classes
    TreeBoosterParams
  53. final def getColsampleBytree: Double
    Definition Classes
    TreeBoosterParams
  54. final def getContribPredictionCol: String

    Definition Classes
    HasContribPredictionCol
  55. final def getCustomEval: EvalTrait
    Definition Classes
    SparkParams
  56. final def getCustomObj: ObjectiveTrait
    Definition Classes
    SparkParams
  57. final def getDefault[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  58. final def getDevice: String
    Definition Classes
    GeneralParams
  59. final def getEta: Double
    Definition Classes
    TreeBoosterParams
  60. final def getEvalMetric: String
    Definition Classes
    LearningTaskParams
  61. final def getExpectileAlpha: String
    Definition Classes
    LearningTaskParams
  62. final def getFeatureNames: Array[String]
    Definition Classes
    SparkParams
  63. final def getFeatureTypes: Array[String]
    Definition Classes
    SparkParams
  64. final def getFeaturesCol: String
    Definition Classes
    HasFeaturesCol
  65. final def getFeaturesCols: Array[String]

    Definition Classes
    HasFeaturesCols
  66. final def getForceRepartition: Boolean
    Definition Classes
    SparkParams
  67. final def getGamma: Double
    Definition Classes
    TreeBoosterParams
  68. final def getGrowPolicy: String
    Definition Classes
    TreeBoosterParams
  69. final def getHuberSlope: Double
    Definition Classes
    LearningTaskParams
  70. final def getInferBatchSize: Int

    Definition Classes
    SparkParams
  71. final def getInteractionConstraints: String
    Definition Classes
    TreeBoosterParams
  72. final def getLabelCol: String
    Definition Classes
    HasLabelCol
  73. final def getLambda: Double
    Definition Classes
    TreeBoosterParams
  74. final def getLambdarankBiasNorm: Double
    Definition Classes
    LearningTaskParams
  75. final def getLambdarankNumPairPerSample: Int
    Definition Classes
    LearningTaskParams
  76. final def getLambdarankPairMethod: String
    Definition Classes
    LearningTaskParams
  77. final def getLambdarankUnbiased: Boolean
    Definition Classes
    LearningTaskParams
  78. final def getLeafPredictionCol: String

    Definition Classes
    HasLeafPredictionCol
  79. final def getMaxBins: Int
    Definition Classes
    TreeBoosterParams
  80. final def getMaxCachedHistNode: Int
    Definition Classes
    TreeBoosterParams
  81. final def getMaxDeltaStep: Double
    Definition Classes
    TreeBoosterParams
  82. final def getMaxDepth: Int
    Definition Classes
    TreeBoosterParams
  83. final def getMaxLeaves: Int
    Definition Classes
    TreeBoosterParams
  84. final def getMaxQuantileBatches: Int
    Definition Classes
    SparkParams
  85. final def getMinCachePageBytes: Long
    Definition Classes
    SparkParams
  86. final def getMinChildWeight: Double
    Definition Classes
    TreeBoosterParams
  87. final def getMissing: Float
    Definition Classes
    SparkParams
  88. final def getMonotoneConstraints: Array[Int]
    Definition Classes
    TreeBoosterParams
  89. final def getNdcgExpGain: Boolean
    Definition Classes
    LearningTaskParams
  90. final def getNormalizeType: String
    Definition Classes
    DartBoosterParams
  91. final def getNthread: Int
    Definition Classes
    GeneralParams
  92. final def getNumClass: Int
    Definition Classes
    LearningTaskParams
  93. final def getNumEarlyStoppingRounds: Int
    Definition Classes
    SparkParams
  94. final def getNumParallelTree: Int
    Definition Classes
    TreeBoosterParams
  95. final def getNumRound: Int
    Definition Classes
    SparkParams
  96. final def getNumWorkers: Int
    Definition Classes
    SparkParams
  97. final def getObjective: String
    Definition Classes
    LearningTaskParams
  98. final def getOneDrop: Boolean
    Definition Classes
    DartBoosterParams
  99. final def getOrDefault[T](param: Param[T]): T
    Definition Classes
    Params
  100. def getParam(paramName: String): Param[Any]
    Definition Classes
    Params
  101. final def getPredictionCol: String
    Definition Classes
    HasPredictionCol
  102. final def getProcessType: String
    Definition Classes
    TreeBoosterParams
  103. final def getRabitTrackerHostIp: String
    Definition Classes
    RabitParams
  104. final def getRabitTrackerPort: Int
    Definition Classes
    RabitParams
  105. final def getRabitTrackerTimeout: Int
    Definition Classes
    RabitParams
  106. final def getRateDrop: Double
    Definition Classes
    DartBoosterParams
  107. final def getRefreshLeaf: Boolean
    Definition Classes
    TreeBoosterParams
  108. final def getSampleType: String
    Definition Classes
    DartBoosterParams
  109. final def getSamplingMethod: String
    Definition Classes
    TreeBoosterParams
  110. final def getScalePosWeight: Double
    Definition Classes
    TreeBoosterParams
  111. final def getSeed: Long
    Definition Classes
    LearningTaskParams
  112. final def getSeedPerIteration: Boolean
    Definition Classes
    LearningTaskParams
  113. final def getSkipDrop: Double
    Definition Classes
    DartBoosterParams
  114. final def getSubsample: Double
    Definition Classes
    TreeBoosterParams
  115. final def getTreeMethod: String
    Definition Classes
    TreeBoosterParams
  116. final def getTweedieVariancePower: Double
    Definition Classes
    LearningTaskParams
  117. final def getUpdater: String
    Definition Classes
    TreeBoosterParams
  118. final def getUseExternalMemory: Boolean
    Definition Classes
    SparkParams
  119. final def getValidateParameters: Boolean
    Definition Classes
    GeneralParams
  120. final def getVerbosity: Int
    Definition Classes
    GeneralParams
  121. final def getWeightCol: String
    Definition Classes
    HasWeightCol
  122. final val growPolicy: Param[String]
    Definition Classes
    TreeBoosterParams
  123. final def hasDefault[T](param: Param[T]): Boolean
    Definition Classes
    Params
  124. def hasParam(paramName: String): Boolean
    Definition Classes
    Params
  125. def hasParent: Boolean
    Definition Classes
    Model
  126. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native() @IntrinsicCandidate()
  127. final val huberSlope: DoubleParam
    Definition Classes
    LearningTaskParams
  128. final val inferBatchSize: IntParam
    Definition Classes
    SparkParams
  129. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  130. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  131. final val interactionConstraints: Param[String]
    Definition Classes
    TreeBoosterParams
  132. final def isDefined(param: Param[_]): Boolean
    Definition Classes
    Params
  133. def isDefinedNonEmpty(param: Param[String]): Boolean
    Definition Classes
    ParamUtils
  134. def isFeaturesColsValid: Boolean

    Check if featuresCols is valid

    Check if featuresCols is valid

    Definition Classes
    HasFeaturesCols
  135. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  136. final def isSet(param: Param[_]): Boolean
    Definition Classes
    Params
  137. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  138. final val labelCol: Param[String]
    Definition Classes
    HasLabelCol
  139. final val lambda: DoubleParam

    L2 regularization term on weights, increase this value will make model more conservative.

    L2 regularization term on weights, increase this value will make model more conservative. [default=1]

    Definition Classes
    TreeBoosterParams
  140. final val lambdarankBiasNorm: DoubleParam
    Definition Classes
    LearningTaskParams
  141. final val lambdarankNumPairPerSample: IntParam
    Definition Classes
    LearningTaskParams
  142. final val lambdarankPairMethod: Param[String]
    Definition Classes
    LearningTaskParams
  143. final val lambdarankUnbiased: BooleanParam
    Definition Classes
    LearningTaskParams
  144. final val leafPredictionCol: Param[String]

    Param for leaf prediction column name.

    Param for leaf prediction column name.

    Definition Classes
    HasLeafPredictionCol
  145. def log: Logger
    Attributes
    protected
    Definition Classes
    Logging
  146. def logDebug(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  147. def logDebug(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  148. def logError(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  149. def logError(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  150. def logInfo(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  151. def logInfo(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  152. def logName: String
    Attributes
    protected
    Definition Classes
    Logging
  153. def logTrace(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  154. def logTrace(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  155. def logWarning(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  156. def logWarning(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  157. final val maxBins: IntParam
    Definition Classes
    TreeBoosterParams
  158. final val maxCachedHistNode: IntParam
    Definition Classes
    TreeBoosterParams
  159. final val maxDeltaStep: DoubleParam
    Definition Classes
    TreeBoosterParams
  160. final val maxDepth: IntParam
    Definition Classes
    TreeBoosterParams
  161. final val maxLeaves: IntParam
    Definition Classes
    TreeBoosterParams
  162. final val maxQuantileBatches: IntParam
    Definition Classes
    SparkParams
  163. final val minCachePageBytes: LongParam
    Definition Classes
    SparkParams
  164. final val minChildWeight: DoubleParam
    Definition Classes
    TreeBoosterParams
  165. final val missing: FloatParam

    the value treated as missing.

    the value treated as missing. default: Float.NaN

    Definition Classes
    SparkParams
  166. final val monotoneConstraints: IntArrayParam
    Definition Classes
    TreeBoosterParams
  167. val nativeBooster: Booster

    Get the native XGBoost Booster

    Get the native XGBoost Booster

    Definition Classes
    XGBoostRegressionModel → XGBoostModel
  168. final val ndcgExpGain: BooleanParam
    Definition Classes
    LearningTaskParams
  169. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  170. lazy val nonXGBoostParams: Array[String]
    Attributes
    protected
    Definition Classes
    NonXGBoostParams
  171. final val normalizeType: Param[String]
    Definition Classes
    DartBoosterParams
  172. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native() @IntrinsicCandidate()
  173. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native() @IntrinsicCandidate()
  174. final val nthread: IntParam
    Definition Classes
    GeneralParams
  175. final val numClass: IntParam
    Definition Classes
    LearningTaskParams
  176. final val numEarlyStoppingRounds: IntParam
    Definition Classes
    SparkParams
  177. def numFeatures: Int
    Definition Classes
    PredictionModel
    Annotations
    @Since( "1.6.0" )
  178. final val numParallelTree: IntParam
    Definition Classes
    TreeBoosterParams
  179. final val numRound: IntParam
    Definition Classes
    SparkParams
  180. final val numWorkers: IntParam
    Definition Classes
    SparkParams
  181. final val objective: Param[String]
    Definition Classes
    LearningTaskParams
  182. final val oneDrop: BooleanParam
    Definition Classes
    DartBoosterParams
  183. lazy val params: Array[Param[_]]
    Definition Classes
    Params
  184. var parent: Estimator[XGBoostRegressionModel]
    Definition Classes
    Model
  185. def postTransform(dataset: Dataset[_], pred: PredictedColumns): Dataset[_]
    Attributes
    protected[spark]
    Definition Classes
    RankerRegressorBaseModel → XGBoostModel
  186. def predict(features: Vector): Double
    Definition Classes
    XGBoostRegressionModel → PredictionModel
  187. def predictSingleInstance(features: Vector): Array[Float]
    Attributes
    protected
    Definition Classes
    XGBoostModel
  188. final val predictionCol: Param[String]
    Definition Classes
    HasPredictionCol
  189. final val processType: Param[String]
    Definition Classes
    TreeBoosterParams
  190. final val rabitTrackerHostIp: Param[String]
    Definition Classes
    RabitParams
  191. final val rabitTrackerPort: IntParam
    Definition Classes
    RabitParams
  192. final val rabitTrackerTimeout: IntParam
    Definition Classes
    RabitParams
  193. final val rateDrop: DoubleParam
    Definition Classes
    DartBoosterParams
  194. final val refreshLeaf: BooleanParam
    Definition Classes
    TreeBoosterParams
  195. final val sampleType: Param[String]
    Definition Classes
    DartBoosterParams
  196. final val samplingMethod: Param[String]
    Definition Classes
    TreeBoosterParams
  197. def save(path: String): Unit
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )
  198. final val scalePosWeight: DoubleParam
    Definition Classes
    TreeBoosterParams
  199. final val seed: LongParam
    Definition Classes
    LearningTaskParams
  200. final val seedPerIteration: BooleanParam
    Definition Classes
    LearningTaskParams
  201. final def set(paramPair: ParamPair[_]): XGBoostRegressionModel.this.type
    Attributes
    protected
    Definition Classes
    Params
  202. final def set(param: String, value: Any): XGBoostRegressionModel.this.type
    Attributes
    protected
    Definition Classes
    Params
  203. final def set[T](param: Param[T], value: T): XGBoostRegressionModel.this.type
    Definition Classes
    Params
  204. def setAftLossDistribution(value: String): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  205. def setAlpha(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  206. def setBaseMarginCol(value: String): XGBoostRegressionModel
    Definition Classes
    SparkParams
  207. def setBaseScore(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  208. def setBooster(value: String): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  209. def setCacheHostRatio(value: Float): XGBoostRegressionModel
    Definition Classes
    SparkParams
  210. def setColsampleBylevel(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  211. def setColsampleBynode(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  212. def setColsampleBytree(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  213. def setContribPredictionCol(value: String): XGBoostRegressionModel
    Definition Classes
    SparkParams
  214. def setCustomEval(value: EvalTrait): XGBoostRegressionModel
    Definition Classes
    SparkParams
  215. def setCustomObj(value: ObjectiveTrait): XGBoostRegressionModel
    Definition Classes
    SparkParams
  216. final def setDefault(paramPairs: ParamPair[_]*): XGBoostRegressionModel.this.type
    Attributes
    protected
    Definition Classes
    Params
  217. final def setDefault[T](param: Param[T], value: T): XGBoostRegressionModel.this.type
    Attributes
    protected[org.apache.spark.ml]
    Definition Classes
    Params
  218. def setDevice(value: String): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  219. def setEta(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  220. def setEvalMetric(value: String): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  221. def setFeatureNames(value: Array[String]): XGBoostRegressionModel
    Definition Classes
    SparkParams
  222. def setFeatureTypes(value: Array[String]): XGBoostRegressionModel
    Definition Classes
    SparkParams
  223. def setFeaturesCol(value: Array[String]): XGBoostRegressionModel
    Definition Classes
    SparkParams
  224. def setFeaturesCol(value: String): XGBoostRegressionModel
    Definition Classes
    PredictionModel
  225. def setForceRepartition(value: Boolean): XGBoostRegressionModel
    Definition Classes
    SparkParams
  226. def setGamma(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  227. def setGrowPolicy(value: String): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  228. def setHuberSlope(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  229. def setInferBatchSize(value: Int): XGBoostRegressionModel
    Definition Classes
    SparkParams
  230. def setInteractionConstraints(value: String): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  231. def setLambda(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  232. def setLambdarankBiasNorm(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  233. def setLambdarankNumPairPerSample(value: Int): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  234. def setLambdarankPairMethod(value: String): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  235. def setLambdarankUnbiased(value: Boolean): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  236. def setLeafPredictionCol(value: String): XGBoostRegressionModel
    Definition Classes
    SparkParams
  237. def setMaxBins(value: Int): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  238. def setMaxCachedHistNode(value: Int): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  239. def setMaxDeltaStep(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  240. def setMaxDepth(value: Int): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  241. def setMaxLeaves(value: Int): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  242. def setMaxQuantileBatches(value: Int): XGBoostRegressionModel
    Definition Classes
    SparkParams
  243. def setMinCachePageBytes(value: Long): XGBoostRegressionModel
    Definition Classes
    SparkParams
  244. def setMinChildWeight(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  245. def setMissing(value: Float): XGBoostRegressionModel
    Definition Classes
    SparkParams
  246. def setNdcgExpGain(value: Boolean): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  247. def setNormalizeType(value: String): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  248. def setNthread(value: Int): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  249. def setNumClass(value: Int): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  250. def setNumEarlyStoppingRounds(value: Int): XGBoostRegressionModel
    Definition Classes
    SparkParams
  251. def setNumParallelTree(value: Int): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  252. def setNumRound(value: Int): XGBoostRegressionModel
    Definition Classes
    SparkParams
  253. def setNumWorkers(value: Int): XGBoostRegressionModel
    Definition Classes
    SparkParams
  254. def setObjective(value: String): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  255. def setOneDrop(value: Boolean): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  256. def setParent(parent: Estimator[XGBoostRegressionModel]): XGBoostRegressionModel
    Definition Classes
    Model
  257. def setPredictionCol(value: String): XGBoostRegressionModel
    Definition Classes
    PredictionModel
  258. def setProcessType(value: String): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  259. def setRabitTrackerHostIp(value: String): XGBoostRegressionModel
    Definition Classes
    SparkParams
  260. def setRabitTrackerPort(value: Int): XGBoostRegressionModel
    Definition Classes
    SparkParams
  261. def setRabitTrackerTimeout(value: Int): XGBoostRegressionModel
    Definition Classes
    SparkParams
  262. def setRateDrop(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  263. def setRefreshLeaf(value: Boolean): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  264. def setSampleType(value: String): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  265. def setSamplingMethod(value: String): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  266. def setScalePosWeight(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  267. def setSeed(value: Long): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  268. def setSeedPerIteration(value: Boolean): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  269. def setSkipDrop(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  270. def setSubsample(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  271. def setTreeMethod(value: String): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  272. def setTweedieVariancePower(value: Double): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  273. def setUpdater(value: String): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  274. def setUseExternalMemory(value: Boolean): XGBoostRegressionModel
    Definition Classes
    SparkParams
  275. def setValidateParameters(value: Boolean): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  276. def setVerbosity(value: Int): XGBoostRegressionModel
    Definition Classes
    XGBoostParams
  277. def setWeightCol(value: String): XGBoostRegressionModel
    Definition Classes
    SparkParams
  278. final val skipDrop: DoubleParam
    Definition Classes
    DartBoosterParams
  279. final val subsample: DoubleParam
    Definition Classes
    TreeBoosterParams
  280. val summary: Option[XGBoostTrainingSummary]
    Definition Classes
    XGBoostRegressionModel → XGBoostModel
  281. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  282. def toString(): String
    Definition Classes
    Identifiable → AnyRef → Any
  283. def transform(dataset: Dataset[_]): DataFrame
    Definition Classes
    XGBoostModel → Transformer
  284. def transform(dataset: Dataset[_], paramMap: ParamMap): DataFrame
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" )
  285. def transform(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): DataFrame
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" ) @varargs()
  286. def transformImpl(dataset: Dataset[_]): DataFrame
    Attributes
    protected
    Definition Classes
    PredictionModel
  287. def transformSchema(schema: StructType): StructType
    Definition Classes
    PredictionModel → PipelineStage
  288. def transformSchema(schema: StructType, logging: Boolean): StructType
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  289. final val treeMethod: Param[String]
    Definition Classes
    TreeBoosterParams
  290. final val tweedieVariancePower: DoubleParam
    Definition Classes
    LearningTaskParams
  291. val uid: String
    Definition Classes
    XGBoostRegressionModel → Identifiable
  292. final val updater: Param[String]
    Definition Classes
    TreeBoosterParams
  293. final val useExternalMemory: BooleanParam
    Definition Classes
    SparkParams
  294. def validateAndTransformSchema(schema: StructType, fitting: Boolean, featuresDataType: DataType): StructType

    XGBoost doesn't use validateAndTransformSchema since spark validateAndTransformSchema needs to ensure the feature is vector type

    XGBoost doesn't use validateAndTransformSchema since spark validateAndTransformSchema needs to ensure the feature is vector type

    Attributes
    protected
    Definition Classes
    XGBPredictorParams → PredictorParams
  295. def validateAndTransformSchema(schema: StructType, fitting: Boolean): StructType
    Definition Classes
    SchemaValidationTrait
  296. def validateFeatureType(schema: StructType): Unit
    Attributes
    protected[spark]
    Definition Classes
    SparkParams
  297. final val validateParameters: BooleanParam
    Definition Classes
    GeneralParams
  298. final val verbosity: IntParam
    Definition Classes
    GeneralParams
  299. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  300. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  301. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  302. final val weightCol: Param[String]
    Definition Classes
    HasWeightCol
  303. def write: MLWriter
    Definition Classes
    XGBoostModel → MLWritable

Deprecated Value Members

  1. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] ) @Deprecated
    Deprecated

Inherited from RankerRegressorBaseModel[XGBoostRegressionModel]

Inherited from XGBPredictorParams[XGBoostRegressionModel]

Inherited from XGBoostModel[XGBoostRegressionModel]

Inherited from ParamUtils[XGBoostRegressionModel]

Inherited from SparkParams[XGBoostRegressionModel]

Inherited from SchemaValidationTrait

Inherited from RabitParams

Inherited from NonXGBoostParams

Inherited from HasContribPredictionCol

Inherited from HasLeafPredictionCol

Inherited from HasWeightCol

Inherited from HasBaseMarginCol

Inherited from HasFeaturesCols

Inherited from XGBoostParams[XGBoostRegressionModel]

Inherited from DartBoosterParams

Inherited from GeneralParams

Inherited from LearningTaskParams

Inherited from TreeBoosterParams

Inherited from MLWritable

Inherited from PredictionModel[Vector, XGBoostRegressionModel]

Inherited from PredictorParams

Inherited from HasPredictionCol

Inherited from HasFeaturesCol

Inherited from HasLabelCol

Inherited from Model[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

getParam

param

Ungrouped