Class/Object

com.microsoft.ml.spark.lightgbm

LightGBMClassifier

Related Docs: object LightGBMClassifier | package lightgbm

Permalink

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

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

Annotations
@InternalWrapper()
Linear Supertypes
LightGBMBase[LightGBMClassificationModel], LightGBMParams, LightGBMPredictionParams, LightGBMLearnerParams, LightGBMBinParams, LightGBMFractionParams, LightGBMSlotParams, LightGBMExecutionParams, HasInitScoreCol, HasValidationIndicatorCol, HasWeightCol, DefaultParamsWritable, MLWritable, Wrappable, HasAdditionalPythonMethods, 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. LightGBMParams
  4. LightGBMPredictionParams
  5. LightGBMLearnerParams
  6. LightGBMBinParams
  7. LightGBMFractionParams
  8. LightGBMSlotParams
  9. LightGBMExecutionParams
  10. HasInitScoreCol
  11. HasValidationIndicatorCol
  12. HasWeightCol
  13. DefaultParamsWritable
  14. MLWritable
  15. Wrappable
  16. HasAdditionalPythonMethods
  17. ProbabilisticClassifier
  18. ProbabilisticClassifierParams
  19. HasThresholds
  20. HasProbabilityCol
  21. Classifier
  22. ClassifierParams
  23. HasRawPredictionCol
  24. Predictor
  25. PredictorParams
  26. HasPredictionCol
  27. HasFeaturesCol
  28. HasLabelCol
  29. Estimator
  30. PipelineStage
  31. Logging
  32. Params
  33. Serializable
  34. Serializable
  35. Identifiable
  36. AnyRef
  37. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new LightGBMClassifier()

    Permalink
  2. new LightGBMClassifier(uid: String)

    Permalink

    uid

    The unique ID.

Value Members

  1. final def !=(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  3. final def $[T](param: Param[T]): T

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  4. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  5. def additionalPythonMethods(): String

    Permalink
    Definition Classes
    HasAdditionalPythonMethods
  6. def afterTrainBatch(batchIndex: Int, dataset: Dataset[_], model: LightGBMClassificationModel): Unit

    Permalink
    Definition Classes
    LightGBMBase
  7. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any
  8. val baggingFraction: DoubleParam

    Permalink
    Definition Classes
    LightGBMFractionParams
  9. val baggingFreq: IntParam

    Permalink
    Definition Classes
    LightGBMParams
  10. val baggingSeed: IntParam

    Permalink
    Definition Classes
    LightGBMParams
  11. def beforeTrainBatch(batchIndex: Int, dataset: Dataset[_], model: Option[LightGBMClassificationModel]): Unit

    Permalink
    Definition Classes
    LightGBMBase
  12. val binSampleCount: IntParam

    Permalink
    Definition Classes
    LightGBMBinParams
  13. val boostFromAverage: BooleanParam

    Permalink
    Definition Classes
    LightGBMParams
  14. val boostingType: Param[String]

    Permalink
    Definition Classes
    LightGBMParams
  15. def castColumns(dataset: Dataset[_], trainingCols: Array[(String, Seq[DataType])]): DataFrame

    Permalink
    Attributes
    protected
    Definition Classes
    LightGBMBase
  16. val categoricalSlotIndexes: IntArrayParam

    Permalink
    Definition Classes
    LightGBMSlotParams
  17. val categoricalSlotNames: StringArrayParam

    Permalink
    Definition Classes
    LightGBMSlotParams
  18. final def clear(param: Param[_]): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    Params
  19. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  20. def copy(extra: ParamMap): LightGBMClassifier

    Permalink
    Definition Classes
    LightGBMClassifier → Predictor → Estimator → PipelineStage → Params
  21. def copyValues[T <: Params](to: T, extra: ParamMap): T

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  22. final def defaultCopy[T <: Params](extra: ParamMap): T

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  23. val defaultListenPort: IntParam

    Permalink
    Definition Classes
    LightGBMExecutionParams
  24. var delegate: Option[LightGBMDelegate]

    Permalink
    Definition Classes
    LightGBMParams
  25. val driverListenPort: IntParam

    Permalink
    Definition Classes
    LightGBMExecutionParams
  26. val earlyStoppingRound: IntParam

    Permalink
    Definition Classes
    LightGBMLearnerParams
  27. final def eq(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  28. def equals(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  29. def explainParam(param: Param[_]): String

    Permalink
    Definition Classes
    Params
  30. def explainParams(): String

    Permalink
    Definition Classes
    Params
  31. def extractLabeledPoints(dataset: Dataset[_], numClasses: Int): RDD[LabeledPoint]

    Permalink
    Attributes
    protected
    Definition Classes
    Classifier
  32. def extractLabeledPoints(dataset: Dataset[_]): RDD[LabeledPoint]

    Permalink
    Attributes
    protected
    Definition Classes
    Predictor
  33. final def extractParamMap(): ParamMap

    Permalink
    Definition Classes
    Params
  34. final def extractParamMap(extra: ParamMap): ParamMap

    Permalink
    Definition Classes
    Params
  35. val featureFraction: DoubleParam

    Permalink
    Definition Classes
    LightGBMFractionParams
  36. final val featuresCol: Param[String]

    Permalink
    Definition Classes
    HasFeaturesCol
  37. val featuresShapCol: Param[String]

    Permalink
    Definition Classes
    LightGBMPredictionParams
  38. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  39. def fit(dataset: Dataset[_]): LightGBMClassificationModel

    Permalink
    Definition Classes
    Predictor → Estimator
  40. def fit(dataset: Dataset[_], paramMaps: Array[ParamMap]): Seq[LightGBMClassificationModel]

    Permalink
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  41. def fit(dataset: Dataset[_], paramMap: ParamMap): LightGBMClassificationModel

    Permalink
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  42. def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): LightGBMClassificationModel

    Permalink
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" ) @varargs()
  43. final def get[T](param: Param[T]): Option[T]

    Permalink
    Definition Classes
    Params
  44. def getBaggingFraction: Double

    Permalink
    Definition Classes
    LightGBMFractionParams
  45. def getBaggingFreq: Int

    Permalink
    Definition Classes
    LightGBMParams
  46. def getBaggingSeed: Int

    Permalink
    Definition Classes
    LightGBMParams
  47. def getBinSampleCount: Int

    Permalink
    Definition Classes
    LightGBMBinParams
  48. def getBoostFromAverage: Boolean

    Permalink
    Definition Classes
    LightGBMParams
  49. def getBoostingType: String

    Permalink
    Definition Classes
    LightGBMParams
  50. def getCategoricalSlotIndexes: Array[Int]

    Permalink
    Definition Classes
    LightGBMSlotParams
  51. def getCategoricalSlotNames: Array[String]

    Permalink
    Definition Classes
    LightGBMSlotParams
  52. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any
  53. final def getDefault[T](param: Param[T]): Option[T]

    Permalink
    Definition Classes
    Params
  54. def getDefaultListenPort: Int

    Permalink
    Definition Classes
    LightGBMExecutionParams
  55. def getDelegate: Option[LightGBMDelegate]

    Permalink
    Definition Classes
    LightGBMParams
  56. def getDriverListenPort: Int

    Permalink
    Definition Classes
    LightGBMExecutionParams
  57. def getEarlyStoppingRound: Int

    Permalink
    Definition Classes
    LightGBMLearnerParams
  58. def getFeatureFraction: Double

    Permalink
    Definition Classes
    LightGBMFractionParams
  59. final def getFeaturesCol: String

    Permalink
    Definition Classes
    HasFeaturesCol
  60. def getFeaturesShapCol: String

    Permalink
    Definition Classes
    LightGBMPredictionParams
  61. def getImprovementTolerance: Double

    Permalink
    Definition Classes
    LightGBMLearnerParams
  62. def getInitScoreCol: String

    Permalink

    Definition Classes
    HasInitScoreCol
  63. def getIsProvideTrainingMetric: Boolean

    Permalink
    Definition Classes
    LightGBMParams
  64. def getIsUnbalance: Boolean

    Permalink
  65. final def getLabelCol: String

    Permalink
    Definition Classes
    HasLabelCol
  66. def getLambdaL1: Double

    Permalink
    Definition Classes
    LightGBMParams
  67. def getLambdaL2: Double

    Permalink
    Definition Classes
    LightGBMParams
  68. def getLeafPredictionCol: String

    Permalink
    Definition Classes
    LightGBMPredictionParams
  69. def getLearningRate: Double

    Permalink
    Definition Classes
    LightGBMParams
  70. def getMaxBin: Int

    Permalink
    Definition Classes
    LightGBMBinParams
  71. def getMaxBinByFeature: Array[Int]

    Permalink
    Definition Classes
    LightGBMParams
  72. def getMaxDeltaStep: Double

    Permalink
    Definition Classes
    LightGBMParams
  73. def getMaxDepth: Int

    Permalink
    Definition Classes
    LightGBMParams
  74. def getMetric: String

    Permalink
    Definition Classes
    LightGBMParams
  75. def getMinDataInLeaf: Int

    Permalink
    Definition Classes
    LightGBMParams
  76. def getMinGainToSplit: Double

    Permalink
    Definition Classes
    LightGBMParams
  77. def getMinSumHessianInLeaf: Double

    Permalink
    Definition Classes
    LightGBMParams
  78. def getModel(trainParams: TrainParams, lightGBMBooster: LightGBMBooster): LightGBMClassificationModel

    Permalink

    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
  79. def getModelString: String

    Permalink
    Definition Classes
    LightGBMParams
  80. def getNegBaggingFraction: Double

    Permalink
    Definition Classes
    LightGBMFractionParams
  81. def getNumBatches: Int

    Permalink
    Definition Classes
    LightGBMExecutionParams
  82. def getNumClasses(dataset: Dataset[_], maxNumClasses: Int): Int

    Permalink
    Attributes
    protected
    Definition Classes
    Classifier
  83. def getNumIterations: Int

    Permalink
    Definition Classes
    LightGBMParams
  84. def getNumLeaves: Int

    Permalink
    Definition Classes
    LightGBMParams
  85. def getNumTasks: Int

    Permalink
    Definition Classes
    LightGBMExecutionParams
  86. def getObjective: String

    Permalink
    Definition Classes
    LightGBMParams
  87. def getOptGroupCol: Option[String]

    Permalink

    Optional group column for Ranking, set to None by default.

    Optional group column for Ranking, set to None by default.

    returns

    None

    Attributes
    protected
    Definition Classes
    LightGBMBase
  88. final def getOrDefault[T](param: Param[T]): T

    Permalink
    Definition Classes
    Params
  89. def getParallelism: String

    Permalink
    Definition Classes
    LightGBMExecutionParams
  90. def getParam(paramName: String): Param[Any]

    Permalink
    Definition Classes
    Params
  91. def getPosBaggingFraction: Double

    Permalink
    Definition Classes
    LightGBMFractionParams
  92. final def getPredictionCol: String

    Permalink
    Definition Classes
    HasPredictionCol
  93. final def getProbabilityCol: String

    Permalink
    Definition Classes
    HasProbabilityCol
  94. final def getRawPredictionCol: String

    Permalink
    Definition Classes
    HasRawPredictionCol
  95. def getRepartitionByGroupingColumn: Boolean

    Permalink
    Definition Classes
    LightGBMExecutionParams
  96. def getSlotNames: Array[String]

    Permalink
    Definition Classes
    LightGBMSlotParams
  97. def getThresholds: Array[Double]

    Permalink
    Definition Classes
    HasThresholds
  98. def getTimeout: Double

    Permalink
    Definition Classes
    LightGBMExecutionParams
  99. def getTopK: Int

    Permalink
    Definition Classes
    LightGBMExecutionParams
  100. def getTrainParams(numTasks: Int, categoricalIndexes: Array[Int], dataset: Dataset[_]): TrainParams

    Permalink

    Gets the training parameters.

    Gets the training parameters.

    returns

    train parameters.

    Definition Classes
    LightGBMClassifierLightGBMBase
  101. def getTrainingCols(): Array[(String, Seq[DataType])]

    Permalink
    Attributes
    protected
    Definition Classes
    LightGBMBase
  102. def getUseBarrierExecutionMode: Boolean

    Permalink
    Definition Classes
    LightGBMExecutionParams
  103. def getValidationIndicatorCol: String

    Permalink

    Definition Classes
    HasValidationIndicatorCol
  104. def getVerbosity: Int

    Permalink
    Definition Classes
    LightGBMParams
  105. def getWeightCol: String

    Permalink

    Definition Classes
    HasWeightCol
  106. final def hasDefault[T](param: Param[T]): Boolean

    Permalink
    Definition Classes
    Params
  107. def hasParam(paramName: String): Boolean

    Permalink
    Definition Classes
    Params
  108. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  109. val improvementTolerance: DoubleParam

    Permalink
    Definition Classes
    LightGBMLearnerParams
  110. val initScoreCol: Param[String]

    Permalink

    The name of the initial score column

    The name of the initial score column

    Definition Classes
    HasInitScoreCol
  111. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  112. def initializeLogIfNecessary(isInterpreter: Boolean): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  113. def innerTrain(dataset: Dataset[_], batchIndex: Int): LightGBMClassificationModel

    Permalink

    Inner train method for LightGBM learners.

    Inner train method for LightGBM learners. Calculates the number of workers, creates a driver thread, and runs mapPartitions on the dataset.

    dataset

    The dataset to train on.

    batchIndex

    In running in batch training mode, gets the batch number.

    returns

    The LightGBM Model from the trained LightGBM Booster.

    Attributes
    protected
    Definition Classes
    LightGBMBase
  114. final def isDefined(param: Param[_]): Boolean

    Permalink
    Definition Classes
    Params
  115. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any
  116. val isProvideTrainingMetric: BooleanParam

    Permalink
    Definition Classes
    LightGBMParams
  117. final def isSet(param: Param[_]): Boolean

    Permalink
    Definition Classes
    Params
  118. def isTraceEnabled(): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  119. val isUnbalance: BooleanParam

    Permalink
  120. final val labelCol: Param[String]

    Permalink
    Definition Classes
    HasLabelCol
  121. val lambdaL1: DoubleParam

    Permalink
    Definition Classes
    LightGBMParams
  122. val lambdaL2: DoubleParam

    Permalink
    Definition Classes
    LightGBMParams
  123. val leafPredictionCol: Param[String]

    Permalink
    Definition Classes
    LightGBMPredictionParams
  124. val learningRate: DoubleParam

    Permalink
    Definition Classes
    LightGBMParams
  125. def log: Logger

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  126. def logDebug(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  127. def logDebug(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  128. def logError(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  129. def logError(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  130. def logInfo(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  131. def logInfo(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  132. def logName: String

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  133. def logTrace(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  134. def logTrace(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  135. def logWarning(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  136. def logWarning(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  137. val maxBin: IntParam

    Permalink
    Definition Classes
    LightGBMBinParams
  138. val maxBinByFeature: IntArrayParam

    Permalink
    Definition Classes
    LightGBMParams
  139. val maxDeltaStep: DoubleParam

    Permalink
    Definition Classes
    LightGBMParams
  140. val maxDepth: IntParam

    Permalink
    Definition Classes
    LightGBMParams
  141. val metric: Param[String]

    Permalink
    Definition Classes
    LightGBMParams
  142. val minDataInLeaf: IntParam

    Permalink
    Definition Classes
    LightGBMParams
  143. val minGainToSplit: DoubleParam

    Permalink
    Definition Classes
    LightGBMParams
  144. val minSumHessianInLeaf: DoubleParam

    Permalink
    Definition Classes
    LightGBMParams
  145. val modelString: Param[String]

    Permalink
    Definition Classes
    LightGBMParams
  146. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  147. val negBaggingFraction: DoubleParam

    Permalink
    Definition Classes
    LightGBMFractionParams
  148. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef
  149. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef
  150. val numBatches: IntParam

    Permalink
    Definition Classes
    LightGBMExecutionParams
  151. val numIterations: IntParam

    Permalink
    Definition Classes
    LightGBMParams
  152. val numLeaves: IntParam

    Permalink
    Definition Classes
    LightGBMParams
  153. val numTasks: IntParam

    Permalink
    Definition Classes
    LightGBMExecutionParams
  154. val objective: Param[String]

    Permalink
    Definition Classes
    LightGBMParams
  155. val parallelism: Param[String]

    Permalink
    Definition Classes
    LightGBMExecutionParams
  156. lazy val params: Array[Param[_]]

    Permalink
    Definition Classes
    Params
  157. val posBaggingFraction: DoubleParam

    Permalink
    Definition Classes
    LightGBMFractionParams
  158. final val predictionCol: Param[String]

    Permalink
    Definition Classes
    HasPredictionCol
  159. def prepareDataframe(dataset: Dataset[_], trainingCols: Array[(String, Seq[DataType])], numTasks: Int): DataFrame

    Permalink
    Attributes
    protected
    Definition Classes
    LightGBMBase
  160. def preprocessData(dataset: DataFrame): DataFrame

    Permalink

    Allow algorithm specific preprocessing of dataset.

    Allow algorithm specific preprocessing of dataset.

    dataset

    The dataset to preprocess prior to training.

    returns

    The preprocessed data.

    Attributes
    protected
    Definition Classes
    LightGBMBase
  161. final val probabilityCol: Param[String]

    Permalink
    Definition Classes
    HasProbabilityCol
  162. final val rawPredictionCol: Param[String]

    Permalink
    Definition Classes
    HasRawPredictionCol
  163. val repartitionByGroupingColumn: BooleanParam

    Permalink
    Definition Classes
    LightGBMExecutionParams
  164. def save(path: String): Unit

    Permalink
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )
  165. final def set(paramPair: ParamPair[_]): LightGBMClassifier.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  166. final def set(param: String, value: Any): LightGBMClassifier.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  167. final def set[T](param: Param[T], value: T): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    Params
  168. def setBaggingFraction(value: Double): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMFractionParams
  169. def setBaggingFreq(value: Int): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  170. def setBaggingSeed(value: Int): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  171. def setBinSampleCount(value: Int): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMBinParams
  172. def setBoostFromAverage(value: Boolean): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  173. def setBoostingType(value: String): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  174. def setCategoricalSlotIndexes(value: Array[Int]): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMSlotParams
  175. def setCategoricalSlotNames(value: Array[String]): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMSlotParams
  176. final def setDefault(paramPairs: ParamPair[_]*): LightGBMClassifier.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  177. final def setDefault[T](param: Param[T], value: T): LightGBMClassifier.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  178. def setDefaultListenPort(value: Int): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  179. def setDelegate(delegate: LightGBMDelegate): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  180. def setDriverListenPort(value: Int): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  181. def setEarlyStoppingRound(value: Int): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMLearnerParams
  182. def setFeatureFraction(value: Double): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMFractionParams
  183. def setFeaturesCol(value: String): LightGBMClassifier

    Permalink
    Definition Classes
    Predictor
  184. def setFeaturesShapCol(value: String): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMPredictionParams
  185. def setImprovementTolerance(value: Double): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMLearnerParams
  186. def setInitScoreCol(value: String): LightGBMClassifier.this.type

    Permalink

    Definition Classes
    HasInitScoreCol
  187. def setIsProvideTrainingMetric(value: Boolean): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  188. def setIsUnbalance(value: Boolean): LightGBMClassifier.this.type

    Permalink
  189. def setLabelCol(value: String): LightGBMClassifier

    Permalink
    Definition Classes
    Predictor
  190. def setLambdaL1(value: Double): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  191. def setLambdaL2(value: Double): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  192. def setLeafPredictionCol(value: String): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMPredictionParams
  193. def setLearningRate(value: Double): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  194. def setMaxBin(value: Int): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMBinParams
  195. def setMaxBinByFeature(value: Array[Int]): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  196. def setMaxDeltaStep(value: Double): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  197. def setMaxDepth(value: Int): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  198. def setMetric(value: String): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  199. def setMinDataInLeaf(value: Int): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  200. def setMinGainToSplit(value: Double): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  201. def setMinSumHessianInLeaf(value: Double): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  202. def setModelString(value: String): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  203. def setNegBaggingFraction(value: Double): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMFractionParams
  204. def setNumBatches(value: Int): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  205. def setNumIterations(value: Int): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  206. def setNumLeaves(value: Int): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  207. def setNumTasks(value: Int): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  208. def setObjective(value: String): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  209. def setParallelism(value: String): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  210. def setPosBaggingFraction(value: Double): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMFractionParams
  211. def setPredictionCol(value: String): LightGBMClassifier

    Permalink
    Definition Classes
    Predictor
  212. def setProbabilityCol(value: String): LightGBMClassifier

    Permalink
    Definition Classes
    ProbabilisticClassifier
  213. def setRawPredictionCol(value: String): LightGBMClassifier

    Permalink
    Definition Classes
    Classifier
  214. def setRepartitionByGroupingColumn(value: Boolean): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  215. def setSlotNames(value: Array[String]): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMSlotParams
  216. def setThresholds(value: Array[Double]): LightGBMClassifier

    Permalink
    Definition Classes
    ProbabilisticClassifier
  217. def setTimeout(value: Double): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  218. def setTopK(value: Int): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  219. def setUseBarrierExecutionMode(value: Boolean): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  220. def setValidationIndicatorCol(value: String): LightGBMClassifier.this.type

    Permalink

    Definition Classes
    HasValidationIndicatorCol
  221. def setVerbosity(value: Int): LightGBMClassifier.this.type

    Permalink
    Definition Classes
    LightGBMParams
  222. def setWeightCol(value: String): LightGBMClassifier.this.type

    Permalink

    Definition Classes
    HasWeightCol
  223. val slotNames: StringArrayParam

    Permalink
    Definition Classes
    LightGBMSlotParams
  224. def stringFromTrainedModel(model: LightGBMClassificationModel): String

    Permalink
    Definition Classes
    LightGBMClassifierLightGBMBase
  225. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef
  226. final val thresholds: DoubleArrayParam

    Permalink
    Definition Classes
    HasThresholds
  227. val timeout: DoubleParam

    Permalink
    Definition Classes
    LightGBMExecutionParams
  228. def toString(): String

    Permalink
    Definition Classes
    Identifiable → AnyRef → Any
  229. val topK: IntParam

    Permalink
    Definition Classes
    LightGBMExecutionParams
  230. def train(dataset: Dataset[_]): LightGBMClassificationModel

    Permalink

    Trains the LightGBM model.

    Trains the LightGBM model. If batches are specified, breaks training dataset into batches for training.

    dataset

    The input dataset to train.

    returns

    The trained model.

    Attributes
    protected
    Definition Classes
    LightGBMBase
  231. def transformSchema(schema: StructType): StructType

    Permalink
    Definition Classes
    Predictor → PipelineStage
  232. def transformSchema(schema: StructType, logging: Boolean): StructType

    Permalink
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  233. val uid: String

    Permalink

    The unique ID.

    The unique ID.

    Definition Classes
    LightGBMClassifier → Identifiable
  234. val useBarrierExecutionMode: BooleanParam

    Permalink
    Definition Classes
    LightGBMExecutionParams
  235. def validateAndTransformSchema(schema: StructType, fitting: Boolean, featuresDataType: DataType): StructType

    Permalink
    Attributes
    protected
    Definition Classes
    ProbabilisticClassifierParams → ClassifierParams → PredictorParams
  236. val validationIndicatorCol: Param[String]

    Permalink

    The name of the validation indicator column

    The name of the validation indicator column

    Definition Classes
    HasValidationIndicatorCol
  237. val verbosity: IntParam

    Permalink
    Definition Classes
    LightGBMParams
  238. final def wait(): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  239. final def wait(arg0: Long, arg1: Int): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  240. final def wait(arg0: Long): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  241. val weightCol: Param[String]

    Permalink

    The name of the weight column

    The name of the weight column

    Definition Classes
    HasWeightCol
  242. def write: MLWriter

    Permalink
    Definition Classes
    DefaultParamsWritable → MLWritable

Inherited from LightGBMParams

Inherited from LightGBMPredictionParams

Inherited from LightGBMLearnerParams

Inherited from LightGBMBinParams

Inherited from LightGBMFractionParams

Inherited from LightGBMSlotParams

Inherited from LightGBMExecutionParams

Inherited from HasInitScoreCol

Inherited from HasValidationIndicatorCol

Inherited from HasWeightCol

Inherited from DefaultParamsWritable

Inherited from MLWritable

Inherited from Wrappable

Inherited from ProbabilisticClassifier[Vector, LightGBMClassifier, LightGBMClassificationModel]

Inherited from ProbabilisticClassifierParams

Inherited from HasThresholds

Inherited from HasProbabilityCol

Inherited from Classifier[Vector, LightGBMClassifier, LightGBMClassificationModel]

Inherited from ClassifierParams

Inherited from HasRawPredictionCol

Inherited from Predictor[Vector, LightGBMClassifier, LightGBMClassificationModel]

Inherited from PredictorParams

Inherited from HasPredictionCol

Inherited from HasFeaturesCol

Inherited from HasLabelCol

Inherited from Estimator[LightGBMClassificationModel]

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

setParam

Ungrouped