Class/Object

com.microsoft.ml.spark.lightgbm

LightGBMRegressor

Related Docs: object LightGBMRegressor | package lightgbm

Permalink

class LightGBMRegressor extends BaseRegressor[Vector, LightGBMRegressor, LightGBMRegressionModel] with LightGBMBase[LightGBMRegressionModel]

Trains a LightGBM Regression 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 Note: The application parameter supports the following values:

Annotations
@InternalWrapper()
Linear Supertypes
LightGBMBase[LightGBMRegressionModel], LightGBMParams, LightGBMPredictionParams, LightGBMLearnerParams, LightGBMBinParams, LightGBMFractionParams, LightGBMSlotParams, LightGBMExecutionParams, HasInitScoreCol, HasValidationIndicatorCol, HasWeightCol, DefaultParamsWritable, MLWritable, Wrappable, HasAdditionalPythonMethods, BaseRegressor[Vector, LightGBMRegressor, LightGBMRegressionModel], Regressor[Vector, LightGBMRegressor, LightGBMRegressionModel], Predictor[Vector, LightGBMRegressor, LightGBMRegressionModel], PredictorParams, HasPredictionCol, HasFeaturesCol, HasLabelCol, Estimator[LightGBMRegressionModel], PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. LightGBMRegressor
  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. BaseRegressor
  18. Regressor
  19. Predictor
  20. PredictorParams
  21. HasPredictionCol
  22. HasFeaturesCol
  23. HasLabelCol
  24. Estimator
  25. PipelineStage
  26. Logging
  27. Params
  28. Serializable
  29. Serializable
  30. Identifiable
  31. AnyRef
  32. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new LightGBMRegressor()

    Permalink
  2. new LightGBMRegressor(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: LightGBMRegressionModel): Unit

    Permalink
    Definition Classes
    LightGBMBase
  7. val alpha: DoubleParam

    Permalink
  8. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any
  9. val baggingFraction: DoubleParam

    Permalink
    Definition Classes
    LightGBMFractionParams
  10. val baggingFreq: IntParam

    Permalink
    Definition Classes
    LightGBMParams
  11. val baggingSeed: IntParam

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

    Permalink
    Definition Classes
    LightGBMBase
  13. val binSampleCount: IntParam

    Permalink
    Definition Classes
    LightGBMBinParams
  14. val boostFromAverage: BooleanParam

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

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

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

    Permalink
    Definition Classes
    LightGBMSlotParams
  18. val categoricalSlotNames: StringArrayParam

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

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

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

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

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

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

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

    Permalink
    Definition Classes
    LightGBMParams
  26. val driverListenPort: IntParam

    Permalink
    Definition Classes
    LightGBMExecutionParams
  27. val earlyStoppingRound: IntParam

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

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

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

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

    Permalink
    Definition Classes
    Params
  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[_]): LightGBMRegressionModel

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

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

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

    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 getAlpha: Double

    Permalink
  45. def getBaggingFraction: Double

    Permalink
    Definition Classes
    LightGBMFractionParams
  46. def getBaggingFreq: Int

    Permalink
    Definition Classes
    LightGBMParams
  47. def getBaggingSeed: Int

    Permalink
    Definition Classes
    LightGBMParams
  48. def getBinSampleCount: Int

    Permalink
    Definition Classes
    LightGBMBinParams
  49. def getBoostFromAverage: Boolean

    Permalink
    Definition Classes
    LightGBMParams
  50. def getBoostingType: String

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

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

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

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

    Permalink
    Definition Classes
    Params
  55. def getDefaultListenPort: Int

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

    Permalink
    Definition Classes
    LightGBMParams
  57. def getDriverListenPort: Int

    Permalink
    Definition Classes
    LightGBMExecutionParams
  58. def getEarlyStoppingRound: Int

    Permalink
    Definition Classes
    LightGBMLearnerParams
  59. def getFeatureFraction: Double

    Permalink
    Definition Classes
    LightGBMFractionParams
  60. final def getFeaturesCol: String

    Permalink
    Definition Classes
    HasFeaturesCol
  61. def getFeaturesShapCol: String

    Permalink
    Definition Classes
    LightGBMPredictionParams
  62. def getImprovementTolerance: Double

    Permalink
    Definition Classes
    LightGBMLearnerParams
  63. def getInitScoreCol: String

    Permalink

    Definition Classes
    HasInitScoreCol
  64. def getIsProvideTrainingMetric: Boolean

    Permalink
    Definition Classes
    LightGBMParams
  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): LightGBMRegressionModel

    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
    LightGBMRegressorLightGBMBase
  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 getNumIterations: Int

    Permalink
    Definition Classes
    LightGBMParams
  83. def getNumLeaves: Int

    Permalink
    Definition Classes
    LightGBMParams
  84. def getNumTasks: Int

    Permalink
    Definition Classes
    LightGBMExecutionParams
  85. def getObjective: String

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

    Permalink
    Definition Classes
    Params
  88. def getParallelism: String

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

    Permalink
    Definition Classes
    Params
  90. def getPosBaggingFraction: Double

    Permalink
    Definition Classes
    LightGBMFractionParams
  91. final def getPredictionCol: String

    Permalink
    Definition Classes
    HasPredictionCol
  92. def getRepartitionByGroupingColumn: Boolean

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

    Permalink
    Definition Classes
    LightGBMSlotParams
  94. def getTimeout: Double

    Permalink
    Definition Classes
    LightGBMExecutionParams
  95. def getTopK: Int

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

    Permalink

    Gets the training parameters.

    Gets the training parameters.

    returns

    train parameters.

    Definition Classes
    LightGBMRegressorLightGBMBase
  97. def getTrainingCols(): Array[(String, Seq[DataType])]

    Permalink
    Attributes
    protected
    Definition Classes
    LightGBMBase
  98. def getTweedieVariancePower: Double

    Permalink
  99. def getUseBarrierExecutionMode: Boolean

    Permalink
    Definition Classes
    LightGBMExecutionParams
  100. def getValidationIndicatorCol: String

    Permalink

    Definition Classes
    HasValidationIndicatorCol
  101. def getVerbosity: Int

    Permalink
    Definition Classes
    LightGBMParams
  102. def getWeightCol: String

    Permalink

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

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

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

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

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

    Permalink

    The name of the initial score column

    The name of the initial score column

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

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

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

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

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

    Permalink
    Definition Classes
    Any
  113. val isProvideTrainingMetric: BooleanParam

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

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

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

    Permalink
    Definition Classes
    HasLabelCol
  117. val lambdaL1: DoubleParam

    Permalink
    Definition Classes
    LightGBMParams
  118. val lambdaL2: DoubleParam

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

    Permalink
    Definition Classes
    LightGBMPredictionParams
  120. val learningRate: DoubleParam

    Permalink
    Definition Classes
    LightGBMParams
  121. def log: Logger

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

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

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

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

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

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

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

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

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

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

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

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

    Permalink
    Definition Classes
    LightGBMBinParams
  134. val maxBinByFeature: IntArrayParam

    Permalink
    Definition Classes
    LightGBMParams
  135. val maxDeltaStep: DoubleParam

    Permalink
    Definition Classes
    LightGBMParams
  136. val maxDepth: IntParam

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

    Permalink
    Definition Classes
    LightGBMParams
  138. val minDataInLeaf: IntParam

    Permalink
    Definition Classes
    LightGBMParams
  139. val minGainToSplit: DoubleParam

    Permalink
    Definition Classes
    LightGBMParams
  140. val minSumHessianInLeaf: DoubleParam

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

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

    Permalink
    Definition Classes
    AnyRef
  143. val negBaggingFraction: DoubleParam

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

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

    Permalink
    Definition Classes
    AnyRef
  146. val numBatches: IntParam

    Permalink
    Definition Classes
    LightGBMExecutionParams
  147. val numIterations: IntParam

    Permalink
    Definition Classes
    LightGBMParams
  148. val numLeaves: IntParam

    Permalink
    Definition Classes
    LightGBMParams
  149. val numTasks: IntParam

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

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

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

    Permalink
    Definition Classes
    Params
  153. val posBaggingFraction: DoubleParam

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

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

    Permalink
    Attributes
    protected
    Definition Classes
    LightGBMBase
  156. 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
  157. val repartitionByGroupingColumn: BooleanParam

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

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

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

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

    Permalink
    Definition Classes
    Params
  162. def setAlpha(value: Double): LightGBMRegressor.this.type

    Permalink
  163. def setBaggingFraction(value: Double): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMFractionParams
  164. def setBaggingFreq(value: Int): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  165. def setBaggingSeed(value: Int): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  166. def setBinSampleCount(value: Int): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMBinParams
  167. def setBoostFromAverage(value: Boolean): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  168. def setBoostingType(value: String): LightGBMRegressor.this.type

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

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

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

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

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

    Permalink
    Definition Classes
    LightGBMExecutionParams
  174. def setDelegate(delegate: LightGBMDelegate): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  175. def setDriverListenPort(value: Int): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  176. def setEarlyStoppingRound(value: Int): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMLearnerParams
  177. def setFeatureFraction(value: Double): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMFractionParams
  178. def setFeaturesCol(value: String): LightGBMRegressor

    Permalink
    Definition Classes
    Predictor
  179. def setFeaturesShapCol(value: String): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMPredictionParams
  180. def setImprovementTolerance(value: Double): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMLearnerParams
  181. def setInitScoreCol(value: String): LightGBMRegressor.this.type

    Permalink

    Definition Classes
    HasInitScoreCol
  182. def setIsProvideTrainingMetric(value: Boolean): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  183. def setLabelCol(value: String): LightGBMRegressor

    Permalink
    Definition Classes
    Predictor
  184. def setLambdaL1(value: Double): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  185. def setLambdaL2(value: Double): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  186. def setLeafPredictionCol(value: String): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMPredictionParams
  187. def setLearningRate(value: Double): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  188. def setMaxBin(value: Int): LightGBMRegressor.this.type

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

    Permalink
    Definition Classes
    LightGBMParams
  190. def setMaxDeltaStep(value: Double): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  191. def setMaxDepth(value: Int): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  192. def setMetric(value: String): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  193. def setMinDataInLeaf(value: Int): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  194. def setMinGainToSplit(value: Double): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  195. def setMinSumHessianInLeaf(value: Double): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  196. def setModelString(value: String): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  197. def setNegBaggingFraction(value: Double): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMFractionParams
  198. def setNumBatches(value: Int): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  199. def setNumIterations(value: Int): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  200. def setNumLeaves(value: Int): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  201. def setNumTasks(value: Int): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  202. def setObjective(value: String): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  203. def setParallelism(value: String): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  204. def setPosBaggingFraction(value: Double): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMFractionParams
  205. def setPredictionCol(value: String): LightGBMRegressor

    Permalink
    Definition Classes
    Predictor
  206. def setRepartitionByGroupingColumn(value: Boolean): LightGBMRegressor.this.type

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

    Permalink
    Definition Classes
    LightGBMSlotParams
  208. def setTimeout(value: Double): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  209. def setTopK(value: Int): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  210. def setTweedieVariancePower(value: Double): LightGBMRegressor.this.type

    Permalink
  211. def setUseBarrierExecutionMode(value: Boolean): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMExecutionParams
  212. def setValidationIndicatorCol(value: String): LightGBMRegressor.this.type

    Permalink

    Definition Classes
    HasValidationIndicatorCol
  213. def setVerbosity(value: Int): LightGBMRegressor.this.type

    Permalink
    Definition Classes
    LightGBMParams
  214. def setWeightCol(value: String): LightGBMRegressor.this.type

    Permalink

    Definition Classes
    HasWeightCol
  215. val slotNames: StringArrayParam

    Permalink
    Definition Classes
    LightGBMSlotParams
  216. def stringFromTrainedModel(model: LightGBMRegressionModel): String

    Permalink
    Definition Classes
    LightGBMRegressorLightGBMBase
  217. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef
  218. val timeout: DoubleParam

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

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

    Permalink
    Definition Classes
    LightGBMExecutionParams
  221. def train(dataset: Dataset[_]): LightGBMRegressionModel

    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
  222. def transformSchema(schema: StructType): StructType

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

    Permalink
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  224. val tweedieVariancePower: DoubleParam

    Permalink
  225. val uid: String

    Permalink

    The unique ID.

    The unique ID.

    Definition Classes
    LightGBMRegressor → Identifiable
  226. val useBarrierExecutionMode: BooleanParam

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

    Permalink
    Attributes
    protected
    Definition Classes
    PredictorParams
  228. val validationIndicatorCol: Param[String]

    Permalink

    The name of the validation indicator column

    The name of the validation indicator column

    Definition Classes
    HasValidationIndicatorCol
  229. val verbosity: IntParam

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

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

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

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

    Permalink

    The name of the weight column

    The name of the weight column

    Definition Classes
    HasWeightCol
  234. 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 Regressor[Vector, LightGBMRegressor, LightGBMRegressionModel]

Inherited from Predictor[Vector, LightGBMRegressor, LightGBMRegressionModel]

Inherited from PredictorParams

Inherited from HasPredictionCol

Inherited from HasFeaturesCol

Inherited from HasLabelCol

Inherited from Estimator[LightGBMRegressionModel]

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