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

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:

  • regression_l2, L2 loss, alias=regression, mean_squared_error, mse, l2_root, root_mean_squared_error, rmse
  • regression_l1, L1 loss, alias=mean_absolute_error, mae
  • huber, Huber loss
  • fair, Fair loss
  • poisson, Poisson regression
  • quantile, Quantile regression
  • mape, MAPE loss, alias=mean_absolute_percentage_error
  • gamma, Gamma regression with log-link. It might be useful, e.g., for modeling insurance claims severity, or for any target that might be gamma-distributed
  • tweedie, Tweedie regression with log-link. It might be useful, e.g., for modeling total loss in insurance, or for any target that might be tweedie-distributed
Linear Supertypes
LightGBMBase[LightGBMRegressionModel], SynapseMLLogging, LightGBMPerformance, ComplexParamsWritable, LightGBMParams, LightGBMCategoricalParams, LightGBMSeedParams, LightGBMObjectiveParams, LightGBMPredictionParams, LightGBMDartParams, LightGBMDatasetParams, LightGBMLearnerParams, LightGBMBinParams, LightGBMFractionParams, LightGBMSlotParams, LightGBMExecutionParams, HasInitScoreCol, HasValidationIndicatorCol, HasWeightCol, DefaultParamsWritable, MLWritable, Wrappable, DotnetWrappable, RWrappable, PythonWrappable, BaseWrappable, 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. SynapseMLLogging
  4. LightGBMPerformance
  5. ComplexParamsWritable
  6. LightGBMParams
  7. LightGBMCategoricalParams
  8. LightGBMSeedParams
  9. LightGBMObjectiveParams
  10. LightGBMPredictionParams
  11. LightGBMDartParams
  12. LightGBMDatasetParams
  13. LightGBMLearnerParams
  14. LightGBMBinParams
  15. LightGBMFractionParams
  16. LightGBMSlotParams
  17. LightGBMExecutionParams
  18. HasInitScoreCol
  19. HasValidationIndicatorCol
  20. HasWeightCol
  21. DefaultParamsWritable
  22. MLWritable
  23. Wrappable
  24. DotnetWrappable
  25. RWrappable
  26. PythonWrappable
  27. BaseWrappable
  28. BaseRegressor
  29. Regressor
  30. Predictor
  31. PredictorParams
  32. HasPredictionCol
  33. HasFeaturesCol
  34. HasLabelCol
  35. Estimator
  36. PipelineStage
  37. Logging
  38. Params
  39. Serializable
  40. Serializable
  41. Identifiable
  42. AnyRef
  43. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

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

    uid

    The unique ID.

Value Members

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

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

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

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

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

    Definition Classes
    LightGBMPerformance
  119. def getPosBaggingFraction: Double
    Definition Classes
    LightGBMFractionParams
  120. def getPredictDisableShapeCheck: Boolean
    Definition Classes
    LightGBMPredictionParams
  121. final def getPredictionCol: String
    Definition Classes
    HasPredictionCol
  122. def getReferenceDataset: Array[Byte]
    Definition Classes
    LightGBMExecutionParams
  123. def getRepartitionByGroupingColumn: Boolean
    Definition Classes
    LightGBMExecutionParams
  124. def getSamplingMode: String
    Definition Classes
    LightGBMExecutionParams
  125. def getSamplingSubsetSize: Int
    Definition Classes
    LightGBMExecutionParams
  126. def getSeed: Int
    Definition Classes
    LightGBMSeedParams
  127. def getSkipDrop: Double
    Definition Classes
    LightGBMDartParams
  128. def getSlotNames: Array[String]
    Definition Classes
    LightGBMSlotParams
  129. def getTimeout: Double
    Definition Classes
    LightGBMExecutionParams
  130. def getTopK: Int
    Definition Classes
    LightGBMExecutionParams
  131. def getTopRate: Double
    Definition Classes
    LightGBMLearnerParams
  132. def getTrainParams(numTasks: Int, featuresSchema: StructField, numTasksPerExec: Int): BaseTrainParams

    Gets the training parameters.

    Gets the training parameters.

    numTasks

    The total number of tasks.

    featuresSchema

    The features column schema.

    numTasksPerExec

    The number of tasks per executor.

    returns

    train parameters.

    Definition Classes
    LightGBMRegressor → LightGBMBase
  133. def getTweedieVariancePower: Double
  134. def getUniformDrop: Boolean
    Definition Classes
    LightGBMDartParams
  135. def getUseBarrierExecutionMode: Boolean
    Definition Classes
    LightGBMExecutionParams
  136. def getUseMissing: Boolean
    Definition Classes
    LightGBMDatasetParams
  137. def getUseSingleDatasetMode: Boolean
    Definition Classes
    LightGBMExecutionParams
  138. def getValidationIndicatorCol: String

    Definition Classes
    HasValidationIndicatorCol
  139. def getVerbosity: Int
    Definition Classes
    LightGBMParams
  140. def getWeightCol: String

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

    The name of the initial score column

    The name of the initial score column

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

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

    Definition Classes
    HasValidationIndicatorCol
  297. def setVerbosity(value: Int): LightGBMRegressor.this.type
    Definition Classes
    LightGBMParams
  298. def setWeightCol(value: String): LightGBMRegressor.this.type

    Definition Classes
    HasWeightCol
  299. def setXGBoostDartMode(value: Boolean): LightGBMRegressor.this.type
    Definition Classes
    LightGBMDartParams
  300. def setZeroAsMissing(value: Boolean): LightGBMRegressor.this.type
    Definition Classes
    LightGBMDatasetParams
  301. val skipDrop: DoubleParam
    Definition Classes
    LightGBMDartParams
  302. val slotNames: StringArrayParam
    Definition Classes
    LightGBMSlotParams
  303. def stringFromTrainedModel(model: LightGBMRegressionModel): String
    Definition Classes
    LightGBMRegressor → LightGBMBase
  304. val timeout: DoubleParam
    Definition Classes
    LightGBMExecutionParams
  305. def toString(): String
    Definition Classes
    Identifiable → AnyRef → Any
  306. val topK: IntParam
    Definition Classes
    LightGBMExecutionParams
  307. val topRate: DoubleParam
    Definition Classes
    LightGBMLearnerParams
  308. def transformSchema(schema: StructType): StructType
    Definition Classes
    Predictor → PipelineStage
  309. val tweedieVariancePower: DoubleParam
  310. val uid: String
    Definition Classes
    LightGBMRegressor → SynapseMLLogging → Identifiable
  311. val uniformDrop: BooleanParam
    Definition Classes
    LightGBMDartParams
  312. val useBarrierExecutionMode: BooleanParam
    Definition Classes
    LightGBMExecutionParams
  313. val useMissing: BooleanParam
    Definition Classes
    LightGBMDatasetParams
  314. val useSingleDatasetMode: BooleanParam
    Definition Classes
    LightGBMExecutionParams
  315. val validationIndicatorCol: Param[String]

    The name of the validation indicator column

    The name of the validation indicator column

    Definition Classes
    HasValidationIndicatorCol
  316. val verbosity: IntParam
    Definition Classes
    LightGBMParams
  317. val weightCol: Param[String]

    The name of the weight column

    The name of the weight column

    Definition Classes
    HasWeightCol
  318. def write: MLWriter
    Definition Classes
    ComplexParamsWritable → MLWritable
  319. val xGBoostDartMode: BooleanParam
    Definition Classes
    LightGBMDartParams
  320. val zeroAsMissing: BooleanParam
    Definition Classes
    LightGBMDatasetParams

Deprecated Value Members

  1. val executionMode: Param[String]
    Definition Classes
    LightGBMExecutionParams
    Annotations
    @deprecated
    Deprecated

    (Since version 0.11.1) Please use 'dataTransferMode'

  2. def setExecutionMode(value: String): LightGBMRegressor.this.type
    Definition Classes
    LightGBMExecutionParams
    Annotations
    @deprecated
    Deprecated

    (Since version 0.11.1) Please use 'setDataTransferMode'