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
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. RWrappable
  25. PythonWrappable
  26. BaseWrappable
  27. BaseRegressor
  28. Regressor
  29. Predictor
  30. PredictorParams
  31. HasPredictionCol
  32. HasFeaturesCol
  33. HasLabelCol
  34. Estimator
  35. PipelineStage
  36. Logging
  37. Params
  38. Serializable
  39. Serializable
  40. Identifiable
  41. AnyRef
  42. 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. val driverListenPort: IntParam
    Definition Classes
    LightGBMExecutionParams
  23. val dropRate: DoubleParam
    Definition Classes
    LightGBMDartParams
  24. val dropSeed: IntParam
    Definition Classes
    LightGBMSeedParams
  25. val earlyStoppingRound: IntParam
    Definition Classes
    LightGBMLearnerParams
  26. def explainParam(param: Param[_]): String
    Definition Classes
    Params
  27. def explainParams(): String
    Definition Classes
    Params
  28. val extraSeed: IntParam
    Definition Classes
    LightGBMSeedParams
  29. final def extractParamMap(): ParamMap
    Definition Classes
    Params
  30. final def extractParamMap(extra: ParamMap): ParamMap
    Definition Classes
    Params
  31. val featureFraction: DoubleParam
    Definition Classes
    LightGBMFractionParams
  32. val featureFractionByNode: DoubleParam
    Definition Classes
    LightGBMFractionParams
  33. val featureFractionSeed: IntParam
    Definition Classes
    LightGBMSeedParams
  34. final val featuresCol: Param[String]
    Definition Classes
    HasFeaturesCol
  35. val featuresShapCol: Param[String]
    Definition Classes
    LightGBMPredictionParams
  36. def fit(dataset: Dataset[_]): LightGBMRegressionModel
    Definition Classes
    Predictor → Estimator
  37. def fit(dataset: Dataset[_], paramMaps: Seq[ParamMap]): Seq[LightGBMRegressionModel]
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  38. def fit(dataset: Dataset[_], paramMap: ParamMap): LightGBMRegressionModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  39. def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): LightGBMRegressionModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" ) @varargs()
  40. val fobj: FObjParam
    Definition Classes
    LightGBMObjectiveParams
  41. final def get[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  42. def getAllPerformanceMeasures: Option[Array[InstrumentationMeasures]]
    Definition Classes
    LightGBMPerformance
  43. def getAlpha: Double
  44. def getBaggingFraction: Double
    Definition Classes
    LightGBMFractionParams
  45. def getBaggingFreq: Int
    Definition Classes
    LightGBMParams
  46. def getBaggingSeed: Int
    Definition Classes
    LightGBMSeedParams
  47. def getBinSampleCount: Int
    Definition Classes
    LightGBMBinParams
  48. def getBoostFromAverage: Boolean
    Definition Classes
    LightGBMParams
  49. def getBoostingType: String
    Definition Classes
    LightGBMParams
  50. def getCatSmooth: Double
    Definition Classes
    LightGBMCategoricalParams
  51. def getCategoricalSlotIndexes: Array[Int]
    Definition Classes
    LightGBMSlotParams
  52. def getCategoricalSlotNames: Array[String]
    Definition Classes
    LightGBMSlotParams
  53. def getCatl2: Double
    Definition Classes
    LightGBMCategoricalParams
  54. def getChunkSize: Int
    Definition Classes
    LightGBMExecutionParams
  55. def getDataRandomSeed: Int
    Definition Classes
    LightGBMSeedParams
  56. def getDataTransferMode: String
    Definition Classes
    LightGBMExecutionParams
  57. final def getDefault[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  58. def getDefaultListenPort: Int
    Definition Classes
    LightGBMExecutionParams
  59. def getDelegate: Option[LightGBMDelegate]
    Definition Classes
    LightGBMParams
  60. def getDeterministic: Boolean
    Definition Classes
    LightGBMSeedParams
  61. def getDriverListenPort: Int
    Definition Classes
    LightGBMExecutionParams
  62. def getDropRate: Double
    Definition Classes
    LightGBMDartParams
  63. def getDropSeed: Int
    Definition Classes
    LightGBMSeedParams
  64. def getEarlyStoppingRound: Int
    Definition Classes
    LightGBMLearnerParams
  65. def getExtraSeed: Int
    Definition Classes
    LightGBMSeedParams
  66. def getFObj: FObjTrait
    Definition Classes
    LightGBMObjectiveParams
  67. def getFeatureFraction: Double
    Definition Classes
    LightGBMFractionParams
  68. def getFeatureFractionByNode: Double
    Definition Classes
    LightGBMFractionParams
  69. def getFeatureFractionSeed: Int
    Definition Classes
    LightGBMSeedParams
  70. final def getFeaturesCol: String
    Definition Classes
    HasFeaturesCol
  71. def getFeaturesShapCol: String
    Definition Classes
    LightGBMPredictionParams
  72. def getImprovementTolerance: Double
    Definition Classes
    LightGBMLearnerParams
  73. def getInitScoreCol: String

    Definition Classes
    HasInitScoreCol
  74. def getIsEnableSparse: Boolean
    Definition Classes
    LightGBMDatasetParams
  75. def getIsProvideTrainingMetric: Boolean
    Definition Classes
    LightGBMParams
  76. final def getLabelCol: String
    Definition Classes
    HasLabelCol
  77. def getLambdaL1: Double
    Definition Classes
    LightGBMParams
  78. def getLambdaL2: Double
    Definition Classes
    LightGBMParams
  79. def getLeafPredictionCol: String
    Definition Classes
    LightGBMPredictionParams
  80. def getLearningRate: Double
    Definition Classes
    LightGBMParams
  81. def getMatrixType: String
    Definition Classes
    LightGBMExecutionParams
  82. def getMaxBin: Int
    Definition Classes
    LightGBMBinParams
  83. def getMaxBinByFeature: Array[Int]
    Definition Classes
    LightGBMParams
  84. def getMaxCatThreshold: Int
    Definition Classes
    LightGBMCategoricalParams
  85. def getMaxCatToOnehot: Int
    Definition Classes
    LightGBMCategoricalParams
  86. def getMaxDeltaStep: Double
    Definition Classes
    LightGBMParams
  87. def getMaxDepth: Int
    Definition Classes
    LightGBMParams
  88. def getMaxDrop: Int
    Definition Classes
    LightGBMDartParams
  89. def getMaxNumClasses: Int
    Definition Classes
    LightGBMParams
  90. def getMaxStreamingOMPThreads: Int
    Definition Classes
    LightGBMExecutionParams
  91. def getMetric: String
    Definition Classes
    LightGBMParams
  92. def getMicroBatchSize: Int
    Definition Classes
    LightGBMExecutionParams
  93. def getMinDataInLeaf: Int
    Definition Classes
    LightGBMParams
  94. def getMinDataPerBin: Int
    Definition Classes
    LightGBMParams
  95. def getMinDataPerGroup: Int
    Definition Classes
    LightGBMCategoricalParams
  96. def getMinGainToSplit: Double
    Definition Classes
    LightGBMParams
  97. def getMinSumHessianInLeaf: Double
    Definition Classes
    LightGBMParams
  98. 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
    LightGBMRegressorLightGBMBase
  99. def getModelString: String
    Definition Classes
    LightGBMParams
  100. def getMonotoneConstraints: Array[Int]
    Definition Classes
    LightGBMLearnerParams
  101. def getMonotoneConstraintsMethod: String
    Definition Classes
    LightGBMLearnerParams
  102. def getMonotonePenalty: Double
    Definition Classes
    LightGBMLearnerParams
  103. def getNegBaggingFraction: Double
    Definition Classes
    LightGBMFractionParams
  104. def getNumBatches: Int
    Definition Classes
    LightGBMExecutionParams
  105. def getNumIterations: Int
    Definition Classes
    LightGBMParams
  106. def getNumLeaves: Int
    Definition Classes
    LightGBMParams
  107. def getNumTasks: Int
    Definition Classes
    LightGBMExecutionParams
  108. def getNumThreads: Int
    Definition Classes
    LightGBMExecutionParams
  109. def getObjective: String
    Definition Classes
    LightGBMObjectiveParams
  110. def getObjectiveSeed: Int
    Definition Classes
    LightGBMSeedParams
  111. final def getOrDefault[T](param: Param[T]): T
    Definition Classes
    Params
  112. def getOtherRate: Double
    Definition Classes
    LightGBMLearnerParams
  113. def getParallelism: String
    Definition Classes
    LightGBMExecutionParams
  114. def getParam(paramName: String): Param[Any]
    Definition Classes
    Params
  115. def getParamInfo(p: Param[_]): ParamInfo[_]
    Definition Classes
    BaseWrappable
  116. def getPassThroughArgs: String
    Definition Classes
    LightGBMExecutionParams
  117. 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
  118. def getPosBaggingFraction: Double
    Definition Classes
    LightGBMFractionParams
  119. def getPredictDisableShapeCheck: Boolean
    Definition Classes
    LightGBMPredictionParams
  120. final def getPredictionCol: String
    Definition Classes
    HasPredictionCol
  121. def getReferenceDataset: Array[Byte]
    Definition Classes
    LightGBMExecutionParams
  122. def getRepartitionByGroupingColumn: Boolean
    Definition Classes
    LightGBMExecutionParams
  123. def getSamplingMode: String
    Definition Classes
    LightGBMExecutionParams
  124. def getSamplingSubsetSize: Int
    Definition Classes
    LightGBMExecutionParams
  125. def getSeed: Int
    Definition Classes
    LightGBMSeedParams
  126. def getSkipDrop: Double
    Definition Classes
    LightGBMDartParams
  127. def getSlotNames: Array[String]
    Definition Classes
    LightGBMSlotParams
  128. def getTimeout: Double
    Definition Classes
    LightGBMExecutionParams
  129. def getTopK: Int
    Definition Classes
    LightGBMExecutionParams
  130. def getTopRate: Double
    Definition Classes
    LightGBMLearnerParams
  131. 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
    LightGBMRegressorLightGBMBase
  132. def getTweedieVariancePower: Double
  133. def getUniformDrop: Boolean
    Definition Classes
    LightGBMDartParams
  134. def getUseBarrierExecutionMode: Boolean
    Definition Classes
    LightGBMExecutionParams
  135. def getUseMissing: Boolean
    Definition Classes
    LightGBMDatasetParams
  136. def getUseSingleDatasetMode: Boolean
    Definition Classes
    LightGBMExecutionParams
  137. def getValidationIndicatorCol: String

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

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

    The name of the initial score column

    The name of the initial score column

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

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

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

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

    The name of the validation indicator column

    The name of the validation indicator column

    Definition Classes
    HasValidationIndicatorCol
  313. val verbosity: IntParam
    Definition Classes
    LightGBMParams
  314. val weightCol: Param[String]

    The name of the weight column

    The name of the weight column

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