Packages

class XGBoostRegressor extends Predictor[Vector, XGBoostRegressor, XGBoostRegressionModel] with XGBoostEstimator[XGBoostRegressor, XGBoostRegressionModel]

Linear Supertypes
XGBoostEstimator[XGBoostRegressor, XGBoostRegressionModel], DefaultParamsWritable, MLWritable, ParamMapConversion, NonParamVariables[XGBoostRegressor, XGBoostRegressionModel], ParamUtils[XGBoostRegressor], SparkParams[XGBoostRegressor], SchemaValidationTrait, RabitParams, NonXGBoostParams, HasContribPredictionCol, HasLeafPredictionCol, HasWeightCol, HasBaseMarginCol, HasFeaturesCols, XGBoostParams[XGBoostRegressor], DartBoosterParams, GeneralParams, LearningTaskParams, TreeBoosterParams, Predictor[Vector, XGBoostRegressor, XGBoostRegressionModel], PredictorParams, HasPredictionCol, HasFeaturesCol, HasLabelCol, Estimator[XGBoostRegressionModel], PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. XGBoostRegressor
  2. XGBoostEstimator
  3. DefaultParamsWritable
  4. MLWritable
  5. ParamMapConversion
  6. NonParamVariables
  7. ParamUtils
  8. SparkParams
  9. SchemaValidationTrait
  10. RabitParams
  11. NonXGBoostParams
  12. HasContribPredictionCol
  13. HasLeafPredictionCol
  14. HasWeightCol
  15. HasBaseMarginCol
  16. HasFeaturesCols
  17. XGBoostParams
  18. DartBoosterParams
  19. GeneralParams
  20. LearningTaskParams
  21. TreeBoosterParams
  22. Predictor
  23. PredictorParams
  24. HasPredictionCol
  25. HasFeaturesCol
  26. HasLabelCol
  27. Estimator
  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 XGBoostRegressor(xgboostParams: Map[String, Any])
  2. new XGBoostRegressor(uid: String)
  3. new XGBoostRegressor()
  4. new XGBoostRegressor(uid: String, xgboostParams: Map[String, Any])

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. def addNonXGBoostParam(ps: Param[_]*): Unit
    Attributes
    protected
    Definition Classes
    NonXGBoostParams
  6. final val aftLossDistribution: Param[String]
    Definition Classes
    LearningTaskParams
  7. final val alpha: DoubleParam
    Definition Classes
    TreeBoosterParams
  8. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  9. 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
  10. final val baseScore: DoubleParam
    Definition Classes
    LearningTaskParams
  11. final val booster: Param[String]
    Definition Classes
    GeneralParams
  12. final val cacheHostRatio: FloatParam
    Definition Classes
    SparkParams
  13. final def clear(param: Param[_]): XGBoostRegressor.this.type
    Definition Classes
    Params
  14. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  15. final val colsampleBylevel: DoubleParam
    Definition Classes
    TreeBoosterParams
  16. final val colsampleBynode: DoubleParam
    Definition Classes
    TreeBoosterParams
  17. final val colsampleBytree: DoubleParam
    Definition Classes
    TreeBoosterParams
  18. final val contribPredictionCol: Param[String]

    Param for contribution prediction column name.

    Param for contribution prediction column name.

    Definition Classes
    HasContribPredictionCol
  19. def copy(extra: ParamMap): XGBoostRegressor
    Definition Classes
    XGBoostEstimator → Estimator → PipelineStage → Params
  20. def copyValues[T <: Params](to: T, extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  21. def createModel(booster: Booster, summary: XGBoostTrainingSummary): XGBoostRegressionModel
    Attributes
    protected
    Definition Classes
    XGBoostRegressor → XGBoostEstimator
  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 finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  40. def fit(dataset: Dataset[_]): XGBoostRegressionModel
    Definition Classes
    Predictor → Estimator
  41. def fit(dataset: Dataset[_], paramMaps: Seq[ParamMap]): Seq[XGBoostRegressionModel]
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  42. def fit(dataset: Dataset[_], paramMap: ParamMap): XGBoostRegressionModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  43. def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): XGBoostRegressionModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" ) @varargs()
  44. final val forceRepartition: BooleanParam
    Definition Classes
    SparkParams
  45. final val gamma: DoubleParam
    Definition Classes
    TreeBoosterParams
  46. final def get[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  47. final def getAftLossDistribution: String
    Definition Classes
    LearningTaskParams
  48. final def getAlpha: Double
    Definition Classes
    TreeBoosterParams
  49. final def getBaseMarginCol: String

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

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

    Definition Classes
    HasFeaturesCols
  69. final def getForceRepartition: Boolean
    Definition Classes
    SparkParams
  70. final def getGamma: Double
    Definition Classes
    TreeBoosterParams
  71. final def getGrowPolicy: String
    Definition Classes
    TreeBoosterParams
  72. final def getHuberSlope: Double
    Definition Classes
    LearningTaskParams
  73. final def getInferBatchSize: Int

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

    Definition Classes
    HasLeafPredictionCol
  82. final def getMaxBins: Int
    Definition Classes
    TreeBoosterParams
  83. final def getMaxCachedHistNode: Int
    Definition Classes
    TreeBoosterParams
  84. final def getMaxDeltaStep: Double
    Definition Classes
    TreeBoosterParams
  85. final def getMaxDepth: Int
    Definition Classes
    TreeBoosterParams
  86. final def getMaxLeaves: Int
    Definition Classes
    TreeBoosterParams
  87. final def getMaxQuantileBatches: Int
    Definition Classes
    SparkParams
  88. final def getMinCachePageBytes: Long
    Definition Classes
    SparkParams
  89. final def getMinChildWeight: Double
    Definition Classes
    TreeBoosterParams
  90. final def getMissing: Float
    Definition Classes
    SparkParams
  91. final def getMonotoneConstraints: Array[Int]
    Definition Classes
    TreeBoosterParams
  92. final def getNdcgExpGain: Boolean
    Definition Classes
    LearningTaskParams
  93. final def getNormalizeType: String
    Definition Classes
    DartBoosterParams
  94. final def getNthread: Int
    Definition Classes
    GeneralParams
  95. final def getNumClass: Int
    Definition Classes
    LearningTaskParams
  96. final def getNumEarlyStoppingRounds: Int
    Definition Classes
    SparkParams
  97. final def getNumParallelTree: Int
    Definition Classes
    TreeBoosterParams
  98. final def getNumRound: Int
    Definition Classes
    SparkParams
  99. final def getNumWorkers: Int
    Definition Classes
    SparkParams
  100. final def getObjective: String
    Definition Classes
    LearningTaskParams
  101. final def getOneDrop: Boolean
    Definition Classes
    DartBoosterParams
  102. final def getOrDefault[T](param: Param[T]): T
    Definition Classes
    Params
  103. def getParam(paramName: String): Param[Any]
    Definition Classes
    Params
  104. final def getPredictionCol: String
    Definition Classes
    HasPredictionCol
  105. final def getProcessType: String
    Definition Classes
    TreeBoosterParams
  106. final def getRabitTrackerHostIp: String
    Definition Classes
    RabitParams
  107. final def getRabitTrackerPort: Int
    Definition Classes
    RabitParams
  108. final def getRabitTrackerTimeout: Int
    Definition Classes
    RabitParams
  109. final def getRateDrop: Double
    Definition Classes
    DartBoosterParams
  110. final def getRefreshLeaf: Boolean
    Definition Classes
    TreeBoosterParams
  111. final def getSampleType: String
    Definition Classes
    DartBoosterParams
  112. final def getSamplingMethod: String
    Definition Classes
    TreeBoosterParams
  113. final def getScalePosWeight: Double
    Definition Classes
    TreeBoosterParams
  114. final def getSeed: Long
    Definition Classes
    LearningTaskParams
  115. final def getSeedPerIteration: Boolean
    Definition Classes
    LearningTaskParams
  116. final def getSkipDrop: Double
    Definition Classes
    DartBoosterParams
  117. final def getSubsample: Double
    Definition Classes
    TreeBoosterParams
  118. final def getTreeMethod: String
    Definition Classes
    TreeBoosterParams
  119. final def getTweedieVariancePower: Double
    Definition Classes
    LearningTaskParams
  120. final def getUpdater: String
    Definition Classes
    TreeBoosterParams
  121. final def getUseExternalMemory: Boolean
    Definition Classes
    SparkParams
  122. final def getValidateParameters: Boolean
    Definition Classes
    GeneralParams
  123. final def getVerbosity: Int
    Definition Classes
    GeneralParams
  124. final def getWeightCol: String
    Definition Classes
    HasWeightCol
  125. def getXGBoostParams: Map[String, Any]

    Convert the user-supplied parameters to the XGBoost parameters.

    Convert the user-supplied parameters to the XGBoost parameters.

    Note that this doesn't contain jvm-specific parameters.

    Definition Classes
    ParamMapConversion
  126. final val growPolicy: Param[String]
    Definition Classes
    TreeBoosterParams
  127. final def hasDefault[T](param: Param[T]): Boolean
    Definition Classes
    Params
  128. def hasParam(paramName: String): Boolean
    Definition Classes
    Params
  129. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  130. final val huberSlope: DoubleParam
    Definition Classes
    LearningTaskParams
  131. final val inferBatchSize: IntParam
    Definition Classes
    SparkParams
  132. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  133. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  134. final val interactionConstraints: Param[String]
    Definition Classes
    TreeBoosterParams
  135. final def isDefined(param: Param[_]): Boolean
    Definition Classes
    Params
  136. def isDefinedNonEmpty(param: Param[String]): Boolean
    Definition Classes
    ParamUtils
  137. def isFeaturesColsValid: Boolean

    Check if featuresCols is valid

    Check if featuresCols is valid

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

    Param for leaf prediction column name.

    Param for leaf prediction column name.

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

    the value treated as missing.

    the value treated as missing. default: Float.NaN

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

    Validate the parameters before training, throw exception if possible

    Validate the parameters before training, throw exception if possible

    Attributes
    protected[spark]
    Definition Classes
    XGBoostRegressor → XGBoostEstimator
  290. def validateAndTransformSchema(schema: StructType, fitting: Boolean, featuresDataType: DataType): StructType
    Attributes
    protected
    Definition Classes
    XGBoostRegressor → PredictorParams
  291. def validateAndTransformSchema(schema: StructType, fitting: Boolean): StructType
    Definition Classes
    SchemaValidationTrait
  292. def validateFeatureType(schema: StructType): Unit
    Attributes
    protected[spark]
    Definition Classes
    SparkParams
  293. final val validateParameters: BooleanParam
    Definition Classes
    GeneralParams
  294. final val verbosity: IntParam
    Definition Classes
    GeneralParams
  295. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  296. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  297. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  298. final val weightCol: Param[String]
    Definition Classes
    HasWeightCol
  299. def write: MLWriter
    Definition Classes
    DefaultParamsWritable → MLWritable
  300. def xgboost2SparkParams(xgboostParams: Map[String, Any]): Unit

    Convert XGBoost parameters to Spark Parameters

    Convert XGBoost parameters to Spark Parameters

    xgboostParams

    XGBoost style parameters

    Definition Classes
    ParamMapConversion

Inherited from XGBoostEstimator[XGBoostRegressor, XGBoostRegressionModel]

Inherited from DefaultParamsWritable

Inherited from MLWritable

Inherited from ParamMapConversion

Inherited from NonParamVariables[XGBoostRegressor, XGBoostRegressionModel]

Inherited from ParamUtils[XGBoostRegressor]

Inherited from SparkParams[XGBoostRegressor]

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[XGBoostRegressor]

Inherited from DartBoosterParams

Inherited from GeneralParams

Inherited from LearningTaskParams

Inherited from TreeBoosterParams

Inherited from Predictor[Vector, XGBoostRegressor, XGBoostRegressionModel]

Inherited from PredictorParams

Inherited from HasPredictionCol

Inherited from HasFeaturesCol

Inherited from HasLabelCol

Inherited from Estimator[XGBoostRegressionModel]

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