Packages

class XGBoostClassificationModel extends ProbabilisticClassificationModel[Vector, XGBoostClassificationModel] with XGBoostModel[XGBoostClassificationModel] with XGBProbabilisticClassifierParams[XGBoostClassificationModel]

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

Instance Constructors

  1. new XGBoostClassificationModel(uid: String)

Value Members

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

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

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

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

    Param for contribution prediction column name.

    Param for contribution prediction column name.

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

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

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

    Definition Classes
    SparkParams
  36. final val featureTypes: StringArrayParam

    Feature types, q is numeric and c is categorical.

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

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

    Param for the names of feature columns.

    Param for the names of feature columns.

    Definition Classes
    HasFeaturesCols
  39. def featuresDataType: DataType
    Attributes
    protected
    Definition Classes
    PredictionModel
  40. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  41. final val forceRepartition: BooleanParam
    Definition Classes
    SparkParams
  42. final val gamma: DoubleParam
    Definition Classes
    TreeBoosterParams
  43. final def get[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  44. final def getAftLossDistribution: String
    Definition Classes
    LearningTaskParams
  45. final def getAlpha: Double
    Definition Classes
    TreeBoosterParams
  46. final def getBaseMarginCol: String

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

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

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

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

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

    Check if featuresCols is valid

    Check if featuresCols is valid

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

    Param for leaf prediction column name.

    Param for leaf prediction column name.

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

    the value treated as missing.

    the value treated as missing. default: Float.NaN

    Definition Classes
    SparkParams
  168. final val monotoneConstraints: IntArrayParam
    Definition Classes
    TreeBoosterParams
  169. val nativeBooster: Booster

    Get the native XGBoost Booster

    Get the native XGBoost Booster

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

Inherited from XGBProbabilisticClassifierParams[XGBoostClassificationModel]

Inherited from XGBoostModel[XGBoostClassificationModel]

Inherited from ParamUtils[XGBoostClassificationModel]

Inherited from SparkParams[XGBoostClassificationModel]

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

Inherited from DartBoosterParams

Inherited from GeneralParams

Inherited from LearningTaskParams

Inherited from TreeBoosterParams

Inherited from MLWritable

Inherited from ProbabilisticClassificationModel[Vector, XGBoostClassificationModel]

Inherited from ProbabilisticClassifierParams

Inherited from HasThresholds

Inherited from HasProbabilityCol

Inherited from ClassificationModel[Vector, XGBoostClassificationModel]

Inherited from ClassifierParams

Inherited from HasRawPredictionCol

Inherited from PredictionModel[Vector, XGBoostClassificationModel]

Inherited from PredictorParams

Inherited from HasPredictionCol

Inherited from HasFeaturesCol

Inherited from HasLabelCol

Inherited from Model[XGBoostClassificationModel]

Inherited from Transformer

Inherited from PipelineStage

Inherited from Logging

Inherited from Params

Inherited from Serializable

Inherited from Serializable

Inherited from Identifiable

Inherited from AnyRef

Inherited from Any

getParam

param

Ungrouped