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

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

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

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

    The name of the validation indicator column

    The name of the validation indicator column

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

    The name of the weight column

    The name of the weight column

    Definition Classes
    HasWeightCol
  317. def write: MLWriter
    Definition Classes
    ComplexParamsWritable → MLWritable
  318. val xGBoostDartMode: BooleanParam
    Definition Classes
    LightGBMDartParams
  319. 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'