Trait

com.microsoft.ml.spark.lightgbm

LightGBMBase

Related Doc: package lightgbm

Permalink

trait LightGBMBase[TrainedModel <: Model[TrainedModel]] extends Estimator[TrainedModel] with LightGBMParams with HasFeaturesCol with HasLabelCol

Linear Supertypes
HasLabelCol, HasFeaturesCol, LightGBMParams, LightGBMPredictionParams, LightGBMLearnerParams, LightGBMBinParams, LightGBMFractionParams, LightGBMSlotParams, LightGBMExecutionParams, HasInitScoreCol, HasValidationIndicatorCol, HasWeightCol, DefaultParamsWritable, MLWritable, Wrappable, HasAdditionalPythonMethods, Estimator[TrainedModel], PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Known Subclasses
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. LightGBMBase
  2. HasLabelCol
  3. HasFeaturesCol
  4. LightGBMParams
  5. LightGBMPredictionParams
  6. LightGBMLearnerParams
  7. LightGBMBinParams
  8. LightGBMFractionParams
  9. LightGBMSlotParams
  10. LightGBMExecutionParams
  11. HasInitScoreCol
  12. HasValidationIndicatorCol
  13. HasWeightCol
  14. DefaultParamsWritable
  15. MLWritable
  16. Wrappable
  17. HasAdditionalPythonMethods
  18. Estimator
  19. PipelineStage
  20. Logging
  21. Params
  22. Serializable
  23. Serializable
  24. Identifiable
  25. AnyRef
  26. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Abstract Value Members

  1. abstract def copy(extra: ParamMap): Estimator[TrainedModel]

    Permalink
    Definition Classes
    Estimator → PipelineStage → Params
  2. abstract def fit(dataset: Dataset[_]): TrainedModel

    Permalink
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  3. abstract def getModel(trainParams: TrainParams, lightGBMBooster: LightGBMBooster): TrainedModel

    Permalink

    Gets the trained model given the train parameters and booster.

    Gets the trained model given the train parameters and booster.

    returns

    trained model.

    Attributes
    protected
  4. abstract def getTrainParams(numTasks: Int, categoricalIndexes: Array[Int], dataset: Dataset[_]): TrainParams

    Permalink

    Gets the training parameters.

    Gets the training parameters.

    returns

    train parameters.

    Attributes
    protected
  5. abstract def stringFromTrainedModel(model: TrainedModel): String

    Permalink
    Attributes
    protected
  6. abstract def transformSchema(schema: StructType): StructType

    Permalink
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  7. abstract val uid: String

    Permalink
    Definition Classes
    Identifiable

Concrete 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: TrainedModel): Unit

    Permalink
  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[TrainedModel]): Unit

    Permalink
  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
  16. val categoricalSlotIndexes: IntArrayParam

    Permalink
    Definition Classes
    LightGBMSlotParams
  17. val categoricalSlotNames: StringArrayParam

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

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

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  20. def copyValues[T <: Params](to: T, extra: ParamMap): T

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

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

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

    Permalink
    Definition Classes
    LightGBMParams
  24. val driverListenPort: IntParam

    Permalink
    Definition Classes
    LightGBMExecutionParams
  25. val earlyStoppingRound: IntParam

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

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

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

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

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

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

    Permalink
    Definition Classes
    Params
  32. val featureFraction: DoubleParam

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

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

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

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  36. def fit(dataset: Dataset[_], paramMaps: Array[ParamMap]): Seq[TrainedModel]

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

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

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

    Permalink
    Definition Classes
    Params
  40. def getBaggingFraction: Double

    Permalink
    Definition Classes
    LightGBMFractionParams
  41. def getBaggingFreq: Int

    Permalink
    Definition Classes
    LightGBMParams
  42. def getBaggingSeed: Int

    Permalink
    Definition Classes
    LightGBMParams
  43. def getBinSampleCount: Int

    Permalink
    Definition Classes
    LightGBMBinParams
  44. def getBoostFromAverage: Boolean

    Permalink
    Definition Classes
    LightGBMParams
  45. def getBoostingType: String

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

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

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

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

    Permalink
    Definition Classes
    Params
  50. def getDefaultListenPort: Int

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

    Permalink
    Definition Classes
    LightGBMParams
  52. def getDriverListenPort: Int

    Permalink
    Definition Classes
    LightGBMExecutionParams
  53. def getEarlyStoppingRound: Int

    Permalink
    Definition Classes
    LightGBMLearnerParams
  54. def getFeatureFraction: Double

    Permalink
    Definition Classes
    LightGBMFractionParams
  55. final def getFeaturesCol: String

    Permalink
    Definition Classes
    HasFeaturesCol
  56. def getFeaturesShapCol: String

    Permalink
    Definition Classes
    LightGBMPredictionParams
  57. def getImprovementTolerance: Double

    Permalink
    Definition Classes
    LightGBMLearnerParams
  58. def getInitScoreCol: String

    Permalink

    Definition Classes
    HasInitScoreCol
  59. def getIsProvideTrainingMetric: Boolean

    Permalink
    Definition Classes
    LightGBMParams
  60. final def getLabelCol: String

    Permalink
    Definition Classes
    HasLabelCol
  61. def getLambdaL1: Double

    Permalink
    Definition Classes
    LightGBMParams
  62. def getLambdaL2: Double

    Permalink
    Definition Classes
    LightGBMParams
  63. def getLeafPredictionCol: String

    Permalink
    Definition Classes
    LightGBMPredictionParams
  64. def getLearningRate: Double

    Permalink
    Definition Classes
    LightGBMParams
  65. def getMaxBin: Int

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

    Permalink
    Definition Classes
    LightGBMParams
  67. def getMaxDeltaStep: Double

    Permalink
    Definition Classes
    LightGBMParams
  68. def getMaxDepth: Int

    Permalink
    Definition Classes
    LightGBMParams
  69. def getMetric: String

    Permalink
    Definition Classes
    LightGBMParams
  70. def getMinDataInLeaf: Int

    Permalink
    Definition Classes
    LightGBMParams
  71. def getMinGainToSplit: Double

    Permalink
    Definition Classes
    LightGBMParams
  72. def getMinSumHessianInLeaf: Double

    Permalink
    Definition Classes
    LightGBMParams
  73. def getModelString: String

    Permalink
    Definition Classes
    LightGBMParams
  74. def getNegBaggingFraction: Double

    Permalink
    Definition Classes
    LightGBMFractionParams
  75. def getNumBatches: Int

    Permalink
    Definition Classes
    LightGBMExecutionParams
  76. def getNumIterations: Int

    Permalink
    Definition Classes
    LightGBMParams
  77. def getNumLeaves: Int

    Permalink
    Definition Classes
    LightGBMParams
  78. def getNumTasks: Int

    Permalink
    Definition Classes
    LightGBMExecutionParams
  79. def getObjective: String

    Permalink
    Definition Classes
    LightGBMParams
  80. 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
  81. final def getOrDefault[T](param: Param[T]): T

    Permalink
    Definition Classes
    Params
  82. def getParallelism: String

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

    Permalink
    Definition Classes
    Params
  84. def getPosBaggingFraction: Double

    Permalink
    Definition Classes
    LightGBMFractionParams
  85. def getRepartitionByGroupingColumn: Boolean

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

    Permalink
    Definition Classes
    LightGBMSlotParams
  87. def getTimeout: Double

    Permalink
    Definition Classes
    LightGBMExecutionParams
  88. def getTopK: Int

    Permalink
    Definition Classes
    LightGBMExecutionParams
  89. def getTrainingCols(): Array[(String, Seq[DataType])]

    Permalink
    Attributes
    protected
  90. def getUseBarrierExecutionMode: Boolean

    Permalink
    Definition Classes
    LightGBMExecutionParams
  91. def getValidationIndicatorCol: String

    Permalink

    Definition Classes
    HasValidationIndicatorCol
  92. def getVerbosity: Int

    Permalink
    Definition Classes
    LightGBMParams
  93. def getWeightCol: String

    Permalink

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

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

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

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

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

    Permalink

    The name of the initial score column

    The name of the initial score column

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

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

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

    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
  102. final def isDefined(param: Param[_]): Boolean

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

    Permalink
    Definition Classes
    Any
  104. val isProvideTrainingMetric: BooleanParam

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

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

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  107. final val labelCol: Param[String]

    Permalink
    Definition Classes
    HasLabelCol
  108. val lambdaL1: DoubleParam

    Permalink
    Definition Classes
    LightGBMParams
  109. val lambdaL2: DoubleParam

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

    Permalink
    Definition Classes
    LightGBMPredictionParams
  111. val learningRate: DoubleParam

    Permalink
    Definition Classes
    LightGBMParams
  112. def log: Logger

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

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

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

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

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

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

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

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

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

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

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

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

    Permalink
    Definition Classes
    LightGBMBinParams
  125. val maxBinByFeature: IntArrayParam

    Permalink
    Definition Classes
    LightGBMParams
  126. val maxDeltaStep: DoubleParam

    Permalink
    Definition Classes
    LightGBMParams
  127. val maxDepth: IntParam

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

    Permalink
    Definition Classes
    LightGBMParams
  129. val minDataInLeaf: IntParam

    Permalink
    Definition Classes
    LightGBMParams
  130. val minGainToSplit: DoubleParam

    Permalink
    Definition Classes
    LightGBMParams
  131. val minSumHessianInLeaf: DoubleParam

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

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

    Permalink
    Definition Classes
    AnyRef
  134. val negBaggingFraction: DoubleParam

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

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

    Permalink
    Definition Classes
    AnyRef
  137. val numBatches: IntParam

    Permalink
    Definition Classes
    LightGBMExecutionParams
  138. val numIterations: IntParam

    Permalink
    Definition Classes
    LightGBMParams
  139. val numLeaves: IntParam

    Permalink
    Definition Classes
    LightGBMParams
  140. val numTasks: IntParam

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

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

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

    Permalink
    Definition Classes
    Params
  144. val posBaggingFraction: DoubleParam

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

    Permalink
    Attributes
    protected
  146. 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
  147. val repartitionByGroupingColumn: BooleanParam

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

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

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

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

    Permalink
    Definition Classes
    Params
  152. def setBaggingFraction(value: Double): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMFractionParams
  153. def setBaggingFreq(value: Int): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  154. def setBaggingSeed(value: Int): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  155. def setBinSampleCount(value: Int): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMBinParams
  156. def setBoostFromAverage(value: Boolean): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  157. def setBoostingType(value: String): LightGBMBase.this.type

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

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

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

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

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

    Permalink
    Definition Classes
    LightGBMExecutionParams
  163. def setDelegate(delegate: LightGBMDelegate): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  164. def setDriverListenPort(value: Int): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  165. def setEarlyStoppingRound(value: Int): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMLearnerParams
  166. def setFeatureFraction(value: Double): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMFractionParams
  167. def setFeaturesShapCol(value: String): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMPredictionParams
  168. def setImprovementTolerance(value: Double): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMLearnerParams
  169. def setInitScoreCol(value: String): LightGBMBase.this.type

    Permalink

    Definition Classes
    HasInitScoreCol
  170. def setIsProvideTrainingMetric(value: Boolean): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  171. def setLambdaL1(value: Double): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  172. def setLambdaL2(value: Double): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  173. def setLeafPredictionCol(value: String): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMPredictionParams
  174. def setLearningRate(value: Double): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  175. def setMaxBin(value: Int): LightGBMBase.this.type

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

    Permalink
    Definition Classes
    LightGBMParams
  177. def setMaxDeltaStep(value: Double): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  178. def setMaxDepth(value: Int): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  179. def setMetric(value: String): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  180. def setMinDataInLeaf(value: Int): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  181. def setMinGainToSplit(value: Double): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  182. def setMinSumHessianInLeaf(value: Double): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  183. def setModelString(value: String): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  184. def setNegBaggingFraction(value: Double): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMFractionParams
  185. def setNumBatches(value: Int): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  186. def setNumIterations(value: Int): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  187. def setNumLeaves(value: Int): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  188. def setNumTasks(value: Int): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  189. def setObjective(value: String): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  190. def setParallelism(value: String): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  191. def setPosBaggingFraction(value: Double): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMFractionParams
  192. def setRepartitionByGroupingColumn(value: Boolean): LightGBMBase.this.type

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

    Permalink
    Definition Classes
    LightGBMSlotParams
  194. def setTimeout(value: Double): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  195. def setTopK(value: Int): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  196. def setUseBarrierExecutionMode(value: Boolean): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  197. def setValidationIndicatorCol(value: String): LightGBMBase.this.type

    Permalink

    Definition Classes
    HasValidationIndicatorCol
  198. def setVerbosity(value: Int): LightGBMBase.this.type

    Permalink
    Definition Classes
    LightGBMParams
  199. def setWeightCol(value: String): LightGBMBase.this.type

    Permalink

    Definition Classes
    HasWeightCol
  200. val slotNames: StringArrayParam

    Permalink
    Definition Classes
    LightGBMSlotParams
  201. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef
  202. val timeout: DoubleParam

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

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

    Permalink
    Definition Classes
    LightGBMExecutionParams
  205. def train(dataset: Dataset[_]): TrainedModel

    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
  206. def transformSchema(schema: StructType, logging: Boolean): StructType

    Permalink
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  207. val useBarrierExecutionMode: BooleanParam

    Permalink
    Definition Classes
    LightGBMExecutionParams
  208. val validationIndicatorCol: Param[String]

    Permalink

    The name of the validation indicator column

    The name of the validation indicator column

    Definition Classes
    HasValidationIndicatorCol
  209. val verbosity: IntParam

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

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

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

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

    Permalink

    The name of the weight column

    The name of the weight column

    Definition Classes
    HasWeightCol
  214. def write: MLWriter

    Permalink
    Definition Classes
    DefaultParamsWritable → MLWritable

Inherited from HasLabelCol

Inherited from HasFeaturesCol

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 Estimator[TrainedModel]

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