Packages

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

Linear Supertypes
RankerRegressorBaseModel[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. XGBoostModel
  4. ParamUtils
  5. SparkParams
  6. SchemaValidationTrait
  7. RabitParams
  8. NonXGBoostParams
  9. HasContribPredictionCol
  10. HasLeafPredictionCol
  11. HasWeightCol
  12. HasBaseMarginCol
  13. HasFeaturesCols
  14. XGBoostParams
  15. DartBoosterParams
  16. GeneralParams
  17. LearningTaskParams
  18. TreeBoosterParams
  19. MLWritable
  20. PredictionModel
  21. PredictorParams
  22. HasPredictionCol
  23. HasFeaturesCol
  24. HasLabelCol
  25. Model
  26. Transformer
  27. PipelineStage
  28. Logging
  29. Params
  30. Serializable
  31. Serializable
  32. Identifiable
  33. AnyRef
  34. 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()
  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. def explainParam(param: Param[_]): String
    Definition Classes
    Params
  31. def explainParams(): String
    Definition Classes
    Params
  32. final def extractParamMap(): ParamMap
    Definition Classes
    Params
  33. final def extractParamMap(extra: ParamMap): ParamMap
    Definition Classes
    Params
  34. def featureIsArrayType(schema: StructType): Boolean
    Attributes
    protected[spark]
    Definition Classes
    SparkParams
  35. 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
  36. 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
  37. final val featuresCol: Param[String]
    Definition Classes
    HasFeaturesCol
  38. final val featuresCols: StringArrayParam

    Param for the names of feature columns.

    Param for the names of feature columns.

    Definition Classes
    HasFeaturesCols
  39. def featuresDataType: DataType
    Attributes
    protected
    Definition Classes
    PredictionModel
  40. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  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()
  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 getFeatureNames: Array[String]
    Definition Classes
    SparkParams
  62. final def getFeatureTypes: Array[String]
    Definition Classes
    SparkParams
  63. final def getFeaturesCol: String
    Definition Classes
    HasFeaturesCol
  64. final def getFeaturesCols: Array[String]

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

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

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

    Check if featuresCols is valid

    Check if featuresCols is valid

    Definition Classes
    HasFeaturesCols
  134. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  135. final def isSet(param: Param[_]): Boolean
    Definition Classes
    Params
  136. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  137. final val labelCol: Param[String]
    Definition Classes
    HasLabelCol
  138. 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
  139. final val lambdarankBiasNorm: DoubleParam
    Definition Classes
    LearningTaskParams
  140. final val lambdarankNumPairPerSample: IntParam
    Definition Classes
    LearningTaskParams
  141. final val lambdarankPairMethod: Param[String]
    Definition Classes
    LearningTaskParams
  142. final val lambdarankUnbiased: BooleanParam
    Definition Classes
    LearningTaskParams
  143. final val leafPredictionCol: Param[String]

    Param for leaf prediction column name.

    Param for leaf prediction column name.

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

    the value treated as missing.

    the value treated as missing. default: Float.NaN

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

    Get the native XGBoost Booster

    Get the native XGBoost Booster

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

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