class LightGBMClassifier extends ProbabilisticClassifier[Vector, LightGBMClassifier, LightGBMClassificationModel] with LightGBMBase[LightGBMClassificationModel] with BasicLogging

Trains a LightGBM Classification model, a fast, distributed, high performance gradient boosting framework based on decision tree algorithms. For more information please see here: https://github.com/Microsoft/LightGBM. For parameter information see here: https://github.com/Microsoft/LightGBM/blob/master/docs/Parameters.rst

Linear Supertypes
LightGBMBase[LightGBMClassificationModel], BasicLogging, LightGBMPerformance, LightGBMParams, LightGBMCategoricalParams, LightGBMSeedParams, LightGBMObjectiveParams, LightGBMPredictionParams, LightGBMDartParams, LightGBMDatasetParams, LightGBMLearnerParams, LightGBMBinParams, LightGBMFractionParams, LightGBMSlotParams, LightGBMExecutionParams, HasInitScoreCol, HasValidationIndicatorCol, HasWeightCol, DefaultParamsWritable, MLWritable, Wrappable, DotnetWrappable, RWrappable, PythonWrappable, BaseWrappable, ProbabilisticClassifier[Vector, LightGBMClassifier, LightGBMClassificationModel], ProbabilisticClassifierParams, HasThresholds, HasProbabilityCol, Classifier[Vector, LightGBMClassifier, LightGBMClassificationModel], ClassifierParams, HasRawPredictionCol, Predictor[Vector, LightGBMClassifier, LightGBMClassificationModel], PredictorParams, HasPredictionCol, HasFeaturesCol, HasLabelCol, Estimator[LightGBMClassificationModel], PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. LightGBMClassifier
  2. LightGBMBase
  3. BasicLogging
  4. LightGBMPerformance
  5. LightGBMParams
  6. LightGBMCategoricalParams
  7. LightGBMSeedParams
  8. LightGBMObjectiveParams
  9. LightGBMPredictionParams
  10. LightGBMDartParams
  11. LightGBMDatasetParams
  12. LightGBMLearnerParams
  13. LightGBMBinParams
  14. LightGBMFractionParams
  15. LightGBMSlotParams
  16. LightGBMExecutionParams
  17. HasInitScoreCol
  18. HasValidationIndicatorCol
  19. HasWeightCol
  20. DefaultParamsWritable
  21. MLWritable
  22. Wrappable
  23. DotnetWrappable
  24. RWrappable
  25. PythonWrappable
  26. BaseWrappable
  27. ProbabilisticClassifier
  28. ProbabilisticClassifierParams
  29. HasThresholds
  30. HasProbabilityCol
  31. Classifier
  32. ClassifierParams
  33. HasRawPredictionCol
  34. Predictor
  35. PredictorParams
  36. HasPredictionCol
  37. HasFeaturesCol
  38. HasLabelCol
  39. Estimator
  40. PipelineStage
  41. Logging
  42. Params
  43. Serializable
  44. Serializable
  45. Identifiable
  46. AnyRef
  47. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new LightGBMClassifier()
  2. new LightGBMClassifier(uid: String)

    uid

    The unique ID.

Value Members

  1. def afterTrainBatch(batchIndex: Int, dataset: Dataset[_], model: LightGBMClassificationModel): Unit
    Definition Classes
    LightGBMBase
  2. val baggingFraction: DoubleParam
    Definition Classes
    LightGBMFractionParams
  3. val baggingFreq: IntParam
    Definition Classes
    LightGBMParams
  4. val baggingSeed: IntParam
    Definition Classes
    LightGBMSeedParams
  5. def beforeTrainBatch(batchIndex: Int, dataset: Dataset[_], model: Option[LightGBMClassificationModel]): Unit
    Definition Classes
    LightGBMBase
  6. val binSampleCount: IntParam
    Definition Classes
    LightGBMBinParams
  7. val boostFromAverage: BooleanParam
    Definition Classes
    LightGBMParams
  8. val boostingType: Param[String]
    Definition Classes
    LightGBMParams
  9. val catSmooth: DoubleParam
    Definition Classes
    LightGBMCategoricalParams
  10. val categoricalSlotIndexes: IntArrayParam
    Definition Classes
    LightGBMSlotParams
  11. val categoricalSlotNames: StringArrayParam
    Definition Classes
    LightGBMSlotParams
  12. val catl2: DoubleParam
    Definition Classes
    LightGBMCategoricalParams
  13. val chunkSize: IntParam
    Definition Classes
    LightGBMExecutionParams
  14. final def clear(param: Param[_]): LightGBMClassifier.this.type
    Definition Classes
    Params
  15. def copy(extra: ParamMap): LightGBMClassifier
    Definition Classes
    LightGBMClassifier → Predictor → Estimator → PipelineStage → Params
  16. val dataRandomSeed: IntParam
    Definition Classes
    LightGBMSeedParams
  17. val defaultListenPort: IntParam
    Definition Classes
    LightGBMExecutionParams
  18. val delegate: Option[LightGBMDelegate]
    Definition Classes
    LightGBMParams
  19. val deterministic: BooleanParam
    Definition Classes
    LightGBMSeedParams
  20. def dotnetAdditionalMethods: String
    Definition Classes
    DotnetWrappable
  21. val driverListenPort: IntParam
    Definition Classes
    LightGBMExecutionParams
  22. val dropRate: DoubleParam
    Definition Classes
    LightGBMDartParams
  23. val dropSeed: IntParam
    Definition Classes
    LightGBMSeedParams
  24. val earlyStoppingRound: IntParam
    Definition Classes
    LightGBMLearnerParams
  25. val executionMode: Param[String]
    Definition Classes
    LightGBMExecutionParams
  26. def explainParam(param: Param[_]): String
    Definition Classes
    Params
  27. def explainParams(): String
    Definition Classes
    Params
  28. val extraSeed: IntParam
    Definition Classes
    LightGBMSeedParams
  29. final def extractParamMap(): ParamMap
    Definition Classes
    Params
  30. final def extractParamMap(extra: ParamMap): ParamMap
    Definition Classes
    Params
  31. val featureFraction: DoubleParam
    Definition Classes
    LightGBMFractionParams
  32. val featureFractionByNode: DoubleParam
    Definition Classes
    LightGBMFractionParams
  33. val featureFractionSeed: IntParam
    Definition Classes
    LightGBMSeedParams
  34. final val featuresCol: Param[String]
    Definition Classes
    HasFeaturesCol
  35. val featuresShapCol: Param[String]
    Definition Classes
    LightGBMPredictionParams
  36. def fit(dataset: Dataset[_]): LightGBMClassificationModel
    Definition Classes
    Predictor → Estimator
  37. def fit(dataset: Dataset[_], paramMaps: Seq[ParamMap]): Seq[LightGBMClassificationModel]
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  38. def fit(dataset: Dataset[_], paramMap: ParamMap): LightGBMClassificationModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  39. def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): LightGBMClassificationModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" ) @varargs()
  40. val fobj: FObjParam
    Definition Classes
    LightGBMObjectiveParams
  41. final def get[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  42. def getAllPerformanceMeasures: Option[Array[InstrumentationMeasures]]
    Definition Classes
    LightGBMPerformance
  43. def getBaggingFraction: Double
    Definition Classes
    LightGBMFractionParams
  44. def getBaggingFreq: Int
    Definition Classes
    LightGBMParams
  45. def getBaggingSeed: Int
    Definition Classes
    LightGBMSeedParams
  46. def getBinSampleCount: Int
    Definition Classes
    LightGBMBinParams
  47. def getBoostFromAverage: Boolean
    Definition Classes
    LightGBMParams
  48. def getBoostingType: String
    Definition Classes
    LightGBMParams
  49. def getCatSmooth: Double
    Definition Classes
    LightGBMCategoricalParams
  50. def getCategoricalSlotIndexes: Array[Int]
    Definition Classes
    LightGBMSlotParams
  51. def getCategoricalSlotNames: Array[String]
    Definition Classes
    LightGBMSlotParams
  52. def getCatl2: Double
    Definition Classes
    LightGBMCategoricalParams
  53. def getChunkSize: Int
    Definition Classes
    LightGBMExecutionParams
  54. def getDataRandomSeed: Int
    Definition Classes
    LightGBMSeedParams
  55. def getDatasetCreationParams(categoricalIndexes: Array[Int], numThreads: Int): String
    Definition Classes
    LightGBMBase
  56. final def getDefault[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  57. def getDefaultListenPort: Int
    Definition Classes
    LightGBMExecutionParams
  58. def getDelegate: Option[LightGBMDelegate]
    Definition Classes
    LightGBMParams
  59. def getDeterministic: Boolean
    Definition Classes
    LightGBMSeedParams
  60. def getDriverListenPort: Int
    Definition Classes
    LightGBMExecutionParams
  61. def getDropRate: Double
    Definition Classes
    LightGBMDartParams
  62. def getDropSeed: Int
    Definition Classes
    LightGBMSeedParams
  63. def getEarlyStoppingRound: Int
    Definition Classes
    LightGBMLearnerParams
  64. def getExecutionMode: String
    Definition Classes
    LightGBMExecutionParams
  65. def getExtraSeed: Int
    Definition Classes
    LightGBMSeedParams
  66. def getFObj: FObjTrait
    Definition Classes
    LightGBMObjectiveParams
  67. def getFeatureFraction: Double
    Definition Classes
    LightGBMFractionParams
  68. def getFeatureFractionByNode: Double
    Definition Classes
    LightGBMFractionParams
  69. def getFeatureFractionSeed: Int
    Definition Classes
    LightGBMSeedParams
  70. final def getFeaturesCol: String
    Definition Classes
    HasFeaturesCol
  71. def getFeaturesShapCol: String
    Definition Classes
    LightGBMPredictionParams
  72. def getImprovementTolerance: Double
    Definition Classes
    LightGBMLearnerParams
  73. def getInitScoreCol: String

    Definition Classes
    HasInitScoreCol
  74. def getIsEnableSparse: Boolean
    Definition Classes
    LightGBMDatasetParams
  75. def getIsProvideTrainingMetric: Boolean
    Definition Classes
    LightGBMParams
  76. def getIsUnbalance: Boolean
  77. final def getLabelCol: String
    Definition Classes
    HasLabelCol
  78. def getLambdaL1: Double
    Definition Classes
    LightGBMParams
  79. def getLambdaL2: Double
    Definition Classes
    LightGBMParams
  80. def getLeafPredictionCol: String
    Definition Classes
    LightGBMPredictionParams
  81. def getLearningRate: Double
    Definition Classes
    LightGBMParams
  82. def getMatrixType: String
    Definition Classes
    LightGBMExecutionParams
  83. def getMaxBin: Int
    Definition Classes
    LightGBMBinParams
  84. def getMaxBinByFeature: Array[Int]
    Definition Classes
    LightGBMParams
  85. def getMaxCatThreshold: Int
    Definition Classes
    LightGBMCategoricalParams
  86. def getMaxCatToOnehot: Int
    Definition Classes
    LightGBMCategoricalParams
  87. def getMaxDeltaStep: Double
    Definition Classes
    LightGBMParams
  88. def getMaxDepth: Int
    Definition Classes
    LightGBMParams
  89. def getMaxDrop: Int
    Definition Classes
    LightGBMDartParams
  90. def getMetric: String
    Definition Classes
    LightGBMParams
  91. def getMicroBatchSize: Int
    Definition Classes
    LightGBMExecutionParams
  92. def getMinDataInLeaf: Int
    Definition Classes
    LightGBMParams
  93. def getMinDataPerBin: Int
    Definition Classes
    LightGBMParams
  94. def getMinDataPerGroup: Int
    Definition Classes
    LightGBMCategoricalParams
  95. def getMinGainToSplit: Double
    Definition Classes
    LightGBMParams
  96. def getMinSumHessianInLeaf: Double
    Definition Classes
    LightGBMParams
  97. def getModel(trainParams: BaseTrainParams, lightGBMBooster: LightGBMBooster): LightGBMClassificationModel

    Gets the trained model given the train parameters and booster.

    Gets the trained model given the train parameters and booster.

    returns

    trained model.

    Definition Classes
    LightGBMClassifierLightGBMBase
  98. def getModelString: String
    Definition Classes
    LightGBMParams
  99. def getMonotoneConstraints: Array[Int]
    Definition Classes
    LightGBMLearnerParams
  100. def getMonotoneConstraintsMethod: String
    Definition Classes
    LightGBMLearnerParams
  101. def getMonotonePenalty: Double
    Definition Classes
    LightGBMLearnerParams
  102. def getNegBaggingFraction: Double
    Definition Classes
    LightGBMFractionParams
  103. def getNumBatches: Int
    Definition Classes
    LightGBMExecutionParams
  104. def getNumIterations: Int
    Definition Classes
    LightGBMParams
  105. def getNumLeaves: Int
    Definition Classes
    LightGBMParams
  106. def getNumTasks: Int
    Definition Classes
    LightGBMExecutionParams
  107. def getNumThreads: Int
    Definition Classes
    LightGBMExecutionParams
  108. def getObjective: String
    Definition Classes
    LightGBMObjectiveParams
  109. def getObjectiveSeed: Int
    Definition Classes
    LightGBMSeedParams
  110. final def getOrDefault[T](param: Param[T]): T
    Definition Classes
    Params
  111. def getOtherRate: Double
    Definition Classes
    LightGBMLearnerParams
  112. def getParallelism: String
    Definition Classes
    LightGBMExecutionParams
  113. def getParam(paramName: String): Param[Any]
    Definition Classes
    Params
  114. def getParamInfo(p: Param[_]): ParamInfo[_]
    Definition Classes
    BaseWrappable
  115. def getPassThroughArgs: String
    Definition Classes
    LightGBMExecutionParams
  116. def getPerformanceMeasures: Option[InstrumentationMeasures]

    In the common case of 1 batch, there is only 1 measure, so this is a convenience method.

    In the common case of 1 batch, there is only 1 measure, so this is a convenience method.

    Definition Classes
    LightGBMPerformance
  117. def getPosBaggingFraction: Double
    Definition Classes
    LightGBMFractionParams
  118. def getPredictDisableShapeCheck: Boolean
    Definition Classes
    LightGBMPredictionParams
  119. final def getPredictionCol: String
    Definition Classes
    HasPredictionCol
  120. final def getProbabilityCol: String
    Definition Classes
    HasProbabilityCol
  121. final def getRawPredictionCol: String
    Definition Classes
    HasRawPredictionCol
  122. def getRepartitionByGroupingColumn: Boolean
    Definition Classes
    LightGBMExecutionParams
  123. def getSeed: Int
    Definition Classes
    LightGBMSeedParams
  124. def getSkipDrop: Double
    Definition Classes
    LightGBMDartParams
  125. def getSlotNames: Array[String]
    Definition Classes
    LightGBMSlotParams
  126. def getSlotNamesWithMetadata(featuresSchema: StructField): Option[Array[String]]
    Definition Classes
    LightGBMBase
  127. def getThresholds: Array[Double]
    Definition Classes
    HasThresholds
  128. def getTimeout: Double
    Definition Classes
    LightGBMExecutionParams
  129. def getTopK: Int
    Definition Classes
    LightGBMExecutionParams
  130. def getTopRate: Double
    Definition Classes
    LightGBMLearnerParams
  131. def getTrainParams(numTasks: Int, featuresSchema: StructField, numTasksPerExec: Int): BaseTrainParams

    Gets the training parameters.

    Gets the training parameters.

    numTasks

    The total number of tasks.

    featuresSchema

    The features column schema.

    numTasksPerExec

    The number of tasks per executor.

    returns

    train parameters.

    Definition Classes
    LightGBMClassifierLightGBMBase
  132. def getUniformDrop: Boolean
    Definition Classes
    LightGBMDartParams
  133. def getUseBarrierExecutionMode: Boolean
    Definition Classes
    LightGBMExecutionParams
  134. def getUseMissing: Boolean
    Definition Classes
    LightGBMDatasetParams
  135. def getUseSingleDatasetMode: Boolean
    Definition Classes
    LightGBMExecutionParams
  136. def getValidationIndicatorCol: String

    Definition Classes
    HasValidationIndicatorCol
  137. def getVerbosity: Int
    Definition Classes
    LightGBMParams
  138. def getWeightCol: String

    Definition Classes
    HasWeightCol
  139. def getXGBoostDartMode: Boolean
    Definition Classes
    LightGBMDartParams
  140. def getZeroAsMissing: Boolean
    Definition Classes
    LightGBMDatasetParams
  141. final def hasDefault[T](param: Param[T]): Boolean
    Definition Classes
    Params
  142. def hasParam(paramName: String): Boolean
    Definition Classes
    Params
  143. val improvementTolerance: DoubleParam
    Definition Classes
    LightGBMLearnerParams
  144. val initScoreCol: Param[String]

    The name of the initial score column

    The name of the initial score column

    Definition Classes
    HasInitScoreCol
  145. final def isDefined(param: Param[_]): Boolean
    Definition Classes
    Params
  146. val isEnableSparse: BooleanParam
    Definition Classes
    LightGBMDatasetParams
  147. val isProvideTrainingMetric: BooleanParam
    Definition Classes
    LightGBMParams
  148. final def isSet(param: Param[_]): Boolean
    Definition Classes
    Params
  149. val isUnbalance: BooleanParam
  150. final val labelCol: Param[String]
    Definition Classes
    HasLabelCol
  151. val lambdaL1: DoubleParam
    Definition Classes
    LightGBMParams
  152. val lambdaL2: DoubleParam
    Definition Classes
    LightGBMParams
  153. val leafPredictionCol: Param[String]
    Definition Classes
    LightGBMPredictionParams
  154. val learningRate: DoubleParam
    Definition Classes
    LightGBMParams
  155. def logClass(): Unit
    Definition Classes
    BasicLogging
  156. def logFit[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  157. def logPredict[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  158. def logTrain[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  159. def logTransform[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  160. def logVerb[T](verb: String, f: ⇒ T): T
    Definition Classes
    BasicLogging
  161. def makeDotnetFile(conf: CodegenConfig): Unit
    Definition Classes
    DotnetWrappable
  162. def makePyFile(conf: CodegenConfig): Unit
    Definition Classes
    PythonWrappable
  163. def makeRFile(conf: CodegenConfig): Unit
    Definition Classes
    RWrappable
  164. val matrixType: Param[String]
    Definition Classes
    LightGBMExecutionParams
  165. val maxBin: IntParam
    Definition Classes
    LightGBMBinParams
  166. val maxBinByFeature: IntArrayParam
    Definition Classes
    LightGBMParams
  167. val maxCatThreshold: IntParam
    Definition Classes
    LightGBMCategoricalParams
  168. val maxCatToOnehot: IntParam
    Definition Classes
    LightGBMCategoricalParams
  169. val maxDeltaStep: DoubleParam
    Definition Classes
    LightGBMParams
  170. val maxDepth: IntParam
    Definition Classes
    LightGBMParams
  171. val maxDrop: IntParam
    Definition Classes
    LightGBMDartParams
  172. val metric: Param[String]
    Definition Classes
    LightGBMParams
  173. val microBatchSize: IntParam
    Definition Classes
    LightGBMExecutionParams
  174. val minDataInLeaf: IntParam
    Definition Classes
    LightGBMParams
  175. val minDataPerBin: IntParam
    Definition Classes
    LightGBMParams
  176. val minDataPerGroup: IntParam
    Definition Classes
    LightGBMCategoricalParams
  177. val minGainToSplit: DoubleParam
    Definition Classes
    LightGBMParams
  178. val minSumHessianInLeaf: DoubleParam
    Definition Classes
    LightGBMParams
  179. val modelString: Param[String]
    Definition Classes
    LightGBMParams
  180. val monotoneConstraints: IntArrayParam
    Definition Classes
    LightGBMLearnerParams
  181. val monotoneConstraintsMethod: Param[String]
    Definition Classes
    LightGBMLearnerParams
  182. val monotonePenalty: DoubleParam
    Definition Classes
    LightGBMLearnerParams
  183. val negBaggingFraction: DoubleParam
    Definition Classes
    LightGBMFractionParams
  184. val numBatches: IntParam
    Definition Classes
    LightGBMExecutionParams
  185. val numIterations: IntParam
    Definition Classes
    LightGBMParams
  186. val numLeaves: IntParam
    Definition Classes
    LightGBMParams
  187. val numTasks: IntParam
    Definition Classes
    LightGBMExecutionParams
  188. val numThreads: IntParam
    Definition Classes
    LightGBMExecutionParams
  189. val objective: Param[String]
    Definition Classes
    LightGBMObjectiveParams
  190. val objectiveSeed: IntParam
    Definition Classes
    LightGBMSeedParams
  191. val otherRate: DoubleParam
    Definition Classes
    LightGBMLearnerParams
  192. val parallelism: Param[String]
    Definition Classes
    LightGBMExecutionParams
  193. lazy val params: Array[Param[_]]
    Definition Classes
    Params
  194. val passThroughArgs: Param[String]
    Definition Classes
    LightGBMExecutionParams
  195. val posBaggingFraction: DoubleParam
    Definition Classes
    LightGBMFractionParams
  196. val predictDisableShapeCheck: BooleanParam
    Definition Classes
    LightGBMPredictionParams
  197. final val predictionCol: Param[String]
    Definition Classes
    HasPredictionCol
  198. final val probabilityCol: Param[String]
    Definition Classes
    HasProbabilityCol
  199. def pyAdditionalMethods: String
    Definition Classes
    PythonWrappable
  200. def pyInitFunc(): String
    Definition Classes
    PythonWrappable
  201. final val rawPredictionCol: Param[String]
    Definition Classes
    HasRawPredictionCol
  202. val repartitionByGroupingColumn: BooleanParam
    Definition Classes
    LightGBMExecutionParams
  203. def save(path: String): Unit
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )
  204. val seed: IntParam
    Definition Classes
    LightGBMSeedParams
  205. final def set[T](param: Param[T], value: T): LightGBMClassifier.this.type
    Definition Classes
    Params
  206. def setBaggingFraction(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMFractionParams
  207. def setBaggingFreq(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  208. def setBaggingSeed(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMSeedParams
  209. def setBinSampleCount(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMBinParams
  210. def setBoostFromAverage(value: Boolean): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  211. def setBoostingType(value: String): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  212. def setCatSmooth(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMCategoricalParams
  213. def setCategoricalSlotIndexes(value: Array[Int]): LightGBMClassifier.this.type
    Definition Classes
    LightGBMSlotParams
  214. def setCategoricalSlotNames(value: Array[String]): LightGBMClassifier.this.type
    Definition Classes
    LightGBMSlotParams
  215. def setCatl2(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMCategoricalParams
  216. def setChunkSize(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  217. def setDataRandomSeed(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMSeedParams
  218. def setDefaultListenPort(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  219. def setDelegate(delegate: LightGBMDelegate): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  220. def setDeterministic(value: Boolean): LightGBMClassifier.this.type
    Definition Classes
    LightGBMSeedParams
  221. def setDriverListenPort(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  222. def setDropRate(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMDartParams
  223. def setDropSeed(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMSeedParams
  224. def setEarlyStoppingRound(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMLearnerParams
  225. def setExecutionMode(value: String): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  226. def setExtraSeed(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMSeedParams
  227. def setFObj(value: FObjTrait): LightGBMClassifier.this.type
    Definition Classes
    LightGBMObjectiveParams
  228. def setFeatureFraction(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMFractionParams
  229. def setFeatureFractionByNode(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMFractionParams
  230. def setFeatureFractionSeed(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMSeedParams
  231. def setFeaturesCol(value: String): LightGBMClassifier
    Definition Classes
    Predictor
  232. def setFeaturesShapCol(value: String): LightGBMClassifier.this.type
    Definition Classes
    LightGBMPredictionParams
  233. def setImprovementTolerance(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMLearnerParams
  234. def setInitScoreCol(value: String): LightGBMClassifier.this.type

    Definition Classes
    HasInitScoreCol
  235. def setIsEnableSparse(value: Boolean): LightGBMClassifier.this.type
    Definition Classes
    LightGBMDatasetParams
  236. def setIsProvideTrainingMetric(value: Boolean): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  237. def setIsUnbalance(value: Boolean): LightGBMClassifier.this.type
  238. def setLabelCol(value: String): LightGBMClassifier
    Definition Classes
    Predictor
  239. def setLambdaL1(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  240. def setLambdaL2(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  241. def setLeafPredictionCol(value: String): LightGBMClassifier.this.type
    Definition Classes
    LightGBMPredictionParams
  242. def setLearningRate(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  243. def setMatrixType(value: String): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  244. def setMaxBin(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMBinParams
  245. def setMaxBinByFeature(value: Array[Int]): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  246. def setMaxCatThreshold(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMCategoricalParams
  247. def setMaxCatToOnehot(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMCategoricalParams
  248. def setMaxDeltaStep(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  249. def setMaxDepth(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  250. def setMaxDrop(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMDartParams
  251. def setMetric(value: String): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  252. def setMicroBatchSize(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  253. def setMinDataInLeaf(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  254. def setMinDataPerBin(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  255. def setMinDataPerGroup(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMCategoricalParams
  256. def setMinGainToSplit(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  257. def setMinSumHessianInLeaf(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  258. def setModelString(value: String): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  259. def setMonotoneConstraints(value: Array[Int]): LightGBMClassifier.this.type
    Definition Classes
    LightGBMLearnerParams
  260. def setMonotoneConstraintsMethod(value: String): LightGBMClassifier.this.type
    Definition Classes
    LightGBMLearnerParams
  261. def setMonotonePenalty(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMLearnerParams
  262. def setNegBaggingFraction(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMFractionParams
  263. def setNumBatches(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  264. def setNumIterations(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  265. def setNumLeaves(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  266. def setNumTasks(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  267. def setNumThreads(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  268. def setObjective(value: String): LightGBMClassifier.this.type
    Definition Classes
    LightGBMObjectiveParams
  269. def setObjectiveSeed(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMSeedParams
  270. def setOtherRate(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMLearnerParams
  271. def setParallelism(value: String): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  272. def setPassThroughArgs(value: String): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  273. def setPosBaggingFraction(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMFractionParams
  274. def setPredictDisableShapeCheck(value: Boolean): LightGBMClassifier.this.type
    Definition Classes
    LightGBMPredictionParams
  275. def setPredictionCol(value: String): LightGBMClassifier
    Definition Classes
    Predictor
  276. def setProbabilityCol(value: String): LightGBMClassifier
    Definition Classes
    ProbabilisticClassifier
  277. def setRawPredictionCol(value: String): LightGBMClassifier
    Definition Classes
    Classifier
  278. def setRepartitionByGroupingColumn(value: Boolean): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  279. def setSeed(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMSeedParams
  280. def setSkipDrop(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMDartParams
  281. def setSlotNames(value: Array[String]): LightGBMClassifier.this.type
    Definition Classes
    LightGBMSlotParams
  282. def setThresholds(value: Array[Double]): LightGBMClassifier
    Definition Classes
    ProbabilisticClassifier
  283. def setTimeout(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  284. def setTopK(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  285. def setTopRate(value: Double): LightGBMClassifier.this.type
    Definition Classes
    LightGBMLearnerParams
  286. def setUniformDrop(value: Boolean): LightGBMClassifier.this.type
    Definition Classes
    LightGBMDartParams
  287. def setUseBarrierExecutionMode(value: Boolean): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  288. def setUseMissing(value: Boolean): LightGBMClassifier.this.type
    Definition Classes
    LightGBMDatasetParams
  289. def setUseSingleDatasetMode(value: Boolean): LightGBMClassifier.this.type
    Definition Classes
    LightGBMExecutionParams
  290. def setValidationIndicatorCol(value: String): LightGBMClassifier.this.type

    Definition Classes
    HasValidationIndicatorCol
  291. def setVerbosity(value: Int): LightGBMClassifier.this.type
    Definition Classes
    LightGBMParams
  292. def setWeightCol(value: String): LightGBMClassifier.this.type

    Definition Classes
    HasWeightCol
  293. def setXGBoostDartMode(value: Boolean): LightGBMClassifier.this.type
    Definition Classes
    LightGBMDartParams
  294. def setZeroAsMissing(value: Boolean): LightGBMClassifier.this.type
    Definition Classes
    LightGBMDatasetParams
  295. val skipDrop: DoubleParam
    Definition Classes
    LightGBMDartParams
  296. val slotNames: StringArrayParam
    Definition Classes
    LightGBMSlotParams
  297. def stringFromTrainedModel(model: LightGBMClassificationModel): String
    Definition Classes
    LightGBMClassifierLightGBMBase
  298. val thresholds: DoubleArrayParam
    Definition Classes
    HasThresholds
  299. val timeout: DoubleParam
    Definition Classes
    LightGBMExecutionParams
  300. def toString(): String
    Definition Classes
    Identifiable → AnyRef → Any
  301. val topK: IntParam
    Definition Classes
    LightGBMExecutionParams
  302. val topRate: DoubleParam
    Definition Classes
    LightGBMLearnerParams
  303. def transformSchema(schema: StructType): StructType
    Definition Classes
    Predictor → PipelineStage
  304. val uid: String
    Definition Classes
    LightGBMClassifierBasicLogging → Identifiable
  305. val uniformDrop: BooleanParam
    Definition Classes
    LightGBMDartParams
  306. val useBarrierExecutionMode: BooleanParam
    Definition Classes
    LightGBMExecutionParams
  307. val useMissing: BooleanParam
    Definition Classes
    LightGBMDatasetParams
  308. val useSingleDatasetMode: BooleanParam
    Definition Classes
    LightGBMExecutionParams
  309. val validationIndicatorCol: Param[String]

    The name of the validation indicator column

    The name of the validation indicator column

    Definition Classes
    HasValidationIndicatorCol
  310. val ver: String
    Definition Classes
    BasicLogging
  311. val verbosity: IntParam
    Definition Classes
    LightGBMParams
  312. val weightCol: Param[String]

    The name of the weight column

    The name of the weight column

    Definition Classes
    HasWeightCol
  313. def write: MLWriter
    Definition Classes
    DefaultParamsWritable → MLWritable
  314. val xGBoostDartMode: BooleanParam
    Definition Classes
    LightGBMDartParams
  315. val zeroAsMissing: BooleanParam
    Definition Classes
    LightGBMDatasetParams