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() @IntrinsicCandidate()
  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. 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 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() @IntrinsicCandidate()
  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 getExpectileAlpha: String
    Definition Classes
    LearningTaskParams
  66. final def getFeatureNames: Array[String]
    Definition Classes
    SparkParams
  67. final def getFeatureTypes: Array[String]
    Definition Classes
    SparkParams
  68. final def getFeaturesCol: String
    Definition Classes
    HasFeaturesCol
  69. final def getFeaturesCols: Array[String]

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

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

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

    Check if featuresCols is valid

    Check if featuresCols is valid

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

    Param for leaf prediction column name.

    Param for leaf prediction column name.

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

    the value treated as missing.

    the value treated as missing. default: Float.NaN

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

Deprecated Value Members

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

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