Packages

class XGBoostClassifier extends ProbabilisticClassifier[Vector, XGBoostClassifier, XGBoostClassificationModel] with XGBoostEstimator[XGBoostClassifier, XGBoostClassificationModel] with XGBProbabilisticClassifierParams[XGBoostClassifier]

Linear Supertypes
XGBProbabilisticClassifierParams[XGBoostClassifier], XGBoostEstimator[XGBoostClassifier, XGBoostClassificationModel], DefaultParamsWritable, MLWritable, ParamMapConversion, NonParamVariables[XGBoostClassifier, XGBoostClassificationModel], ParamUtils[XGBoostClassifier], SparkParams[XGBoostClassifier], SchemaValidationTrait, RabitParams, NonXGBoostParams, HasContribPredictionCol, HasLeafPredictionCol, HasWeightCol, HasBaseMarginCol, HasFeaturesCols, XGBoostParams[XGBoostClassifier], DartBoosterParams, GeneralParams, LearningTaskParams, TreeBoosterParams, ProbabilisticClassifier[Vector, XGBoostClassifier, XGBoostClassificationModel], ProbabilisticClassifierParams, HasThresholds, HasProbabilityCol, Classifier[Vector, XGBoostClassifier, XGBoostClassificationModel], ClassifierParams, HasRawPredictionCol, Predictor[Vector, XGBoostClassifier, XGBoostClassificationModel], PredictorParams, HasPredictionCol, HasFeaturesCol, HasLabelCol, Estimator[XGBoostClassificationModel], PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. XGBoostClassifier
  2. XGBProbabilisticClassifierParams
  3. XGBoostEstimator
  4. DefaultParamsWritable
  5. MLWritable
  6. ParamMapConversion
  7. NonParamVariables
  8. ParamUtils
  9. SparkParams
  10. SchemaValidationTrait
  11. RabitParams
  12. NonXGBoostParams
  13. HasContribPredictionCol
  14. HasLeafPredictionCol
  15. HasWeightCol
  16. HasBaseMarginCol
  17. HasFeaturesCols
  18. XGBoostParams
  19. DartBoosterParams
  20. GeneralParams
  21. LearningTaskParams
  22. TreeBoosterParams
  23. ProbabilisticClassifier
  24. ProbabilisticClassifierParams
  25. HasThresholds
  26. HasProbabilityCol
  27. Classifier
  28. ClassifierParams
  29. HasRawPredictionCol
  30. Predictor
  31. PredictorParams
  32. HasPredictionCol
  33. HasFeaturesCol
  34. HasLabelCol
  35. Estimator
  36. PipelineStage
  37. Logging
  38. Params
  39. Serializable
  40. Serializable
  41. Identifiable
  42. AnyRef
  43. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new XGBoostClassifier(xgboostParams: Map[String, Any])
  2. new XGBoostClassifier(uid: String)
  3. new XGBoostClassifier()
  4. new XGBoostClassifier(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[_]): XGBoostClassifier.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): XGBoostClassifier
    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): XGBoostClassificationModel
    Attributes
    protected
    Definition Classes
    XGBoostClassifier → 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[_]): XGBoostClassificationModel
    Definition Classes
    Predictor → Estimator
  41. def fit(dataset: Dataset[_], paramMaps: Seq[ParamMap]): Seq[XGBoostClassificationModel]
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  42. def fit(dataset: Dataset[_], paramMap: ParamMap): XGBoostClassificationModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  43. def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): XGBoostClassificationModel
    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. def getNumClasses(dataset: Dataset[_], maxNumClasses: Int): Int
    Attributes
    protected
    Definition Classes
    Classifier
  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 getProbabilityCol: String
    Definition Classes
    HasProbabilityCol
  107. final def getProcessType: String
    Definition Classes
    TreeBoosterParams
  108. final def getRabitTrackerHostIp: String
    Definition Classes
    RabitParams
  109. final def getRabitTrackerPort: Int
    Definition Classes
    RabitParams
  110. final def getRabitTrackerTimeout: Int
    Definition Classes
    RabitParams
  111. final def getRateDrop: Double
    Definition Classes
    DartBoosterParams
  112. final def getRawPredictionCol: String
    Definition Classes
    HasRawPredictionCol
  113. final def getRefreshLeaf: Boolean
    Definition Classes
    TreeBoosterParams
  114. final def getSampleType: String
    Definition Classes
    DartBoosterParams
  115. final def getSamplingMethod: String
    Definition Classes
    TreeBoosterParams
  116. final def getScalePosWeight: Double
    Definition Classes
    TreeBoosterParams
  117. final def getSeed: Long
    Definition Classes
    LearningTaskParams
  118. final def getSeedPerIteration: Boolean
    Definition Classes
    LearningTaskParams
  119. final def getSkipDrop: Double
    Definition Classes
    DartBoosterParams
  120. final def getSubsample: Double
    Definition Classes
    TreeBoosterParams
  121. def getThresholds: Array[Double]
    Definition Classes
    HasThresholds
  122. final def getTreeMethod: String
    Definition Classes
    TreeBoosterParams
  123. final def getTweedieVariancePower: Double
    Definition Classes
    LearningTaskParams
  124. final def getUpdater: String
    Definition Classes
    TreeBoosterParams
  125. final def getUseExternalMemory: Boolean
    Definition Classes
    SparkParams
  126. final def getValidateParameters: Boolean
    Definition Classes
    GeneralParams
  127. final def getVerbosity: Int
    Definition Classes
    GeneralParams
  128. final def getWeightCol: String
    Definition Classes
    HasWeightCol
  129. 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
  130. final val growPolicy: Param[String]
    Definition Classes
    TreeBoosterParams
  131. final def hasDefault[T](param: Param[T]): Boolean
    Definition Classes
    Params
  132. def hasParam(paramName: String): Boolean
    Definition Classes
    Params
  133. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  134. final val huberSlope: DoubleParam
    Definition Classes
    LearningTaskParams
  135. final val inferBatchSize: IntParam
    Definition Classes
    SparkParams
  136. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  137. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  138. final val interactionConstraints: Param[String]
    Definition Classes
    TreeBoosterParams
  139. final def isDefined(param: Param[_]): Boolean
    Definition Classes
    Params
  140. def isDefinedNonEmpty(param: Param[String]): Boolean
    Definition Classes
    ParamUtils
  141. def isFeaturesColsValid: Boolean

    Check if featuresCols is valid

    Check if featuresCols is valid

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

    Param for leaf prediction column name.

    Param for leaf prediction column name.

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

    the value treated as missing.

    the value treated as missing. default: Float.NaN

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

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

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

    Attributes
    protected
    Definition Classes
    XGBProbabilisticClassifierParams → ProbabilisticClassifierParams → ClassifierParams → PredictorParams
  301. def validateAndTransformSchema(schema: StructType, fitting: Boolean): StructType
    Definition Classes
    SchemaValidationTrait
  302. def validateFeatureType(schema: StructType): Unit
    Attributes
    protected[spark]
    Definition Classes
    SparkParams
  303. final val validateParameters: BooleanParam
    Definition Classes
    GeneralParams
  304. final val verbosity: IntParam
    Definition Classes
    GeneralParams
  305. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  306. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  307. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  308. final val weightCol: Param[String]
    Definition Classes
    HasWeightCol
  309. def write: MLWriter
    Definition Classes
    DefaultParamsWritable → MLWritable
  310. 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 XGBProbabilisticClassifierParams[XGBoostClassifier]

Inherited from XGBoostEstimator[XGBoostClassifier, XGBoostClassificationModel]

Inherited from DefaultParamsWritable

Inherited from MLWritable

Inherited from ParamMapConversion

Inherited from NonParamVariables[XGBoostClassifier, XGBoostClassificationModel]

Inherited from ParamUtils[XGBoostClassifier]

Inherited from SparkParams[XGBoostClassifier]

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

Inherited from DartBoosterParams

Inherited from GeneralParams

Inherited from LearningTaskParams

Inherited from TreeBoosterParams

Inherited from ProbabilisticClassifier[Vector, XGBoostClassifier, XGBoostClassificationModel]

Inherited from ProbabilisticClassifierParams

Inherited from HasThresholds

Inherited from HasProbabilityCol

Inherited from Classifier[Vector, XGBoostClassifier, XGBoostClassificationModel]

Inherited from ClassifierParams

Inherited from HasRawPredictionCol

Inherited from Predictor[Vector, XGBoostClassifier, XGBoostClassificationModel]

Inherited from PredictorParams

Inherited from HasPredictionCol

Inherited from HasFeaturesCol

Inherited from HasLabelCol

Inherited from Estimator[XGBoostClassificationModel]

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