Packages

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

Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. LightGBMBase
  2. BasicLogging
  3. HasLabelCol
  4. HasFeaturesCol
  5. LightGBMParams
  6. LightGBMObjectiveParams
  7. LightGBMPredictionParams
  8. LightGBMDartParams
  9. LightGBMLearnerParams
  10. LightGBMBinParams
  11. LightGBMFractionParams
  12. LightGBMSlotParams
  13. LightGBMExecutionParams
  14. HasInitScoreCol
  15. HasValidationIndicatorCol
  16. HasWeightCol
  17. DefaultParamsWritable
  18. MLWritable
  19. Wrappable
  20. RWrappable
  21. PythonWrappable
  22. BaseWrappable
  23. Estimator
  24. PipelineStage
  25. Logging
  26. Params
  27. Serializable
  28. Serializable
  29. Identifiable
  30. AnyRef
  31. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Abstract Value Members

  1. abstract def copy(extra: ParamMap): Estimator[TrainedModel]
    Definition Classes
    Estimator → PipelineStage → Params
  2. abstract def fit(dataset: Dataset[_]): TrainedModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  3. abstract def getModel(trainParams: TrainParams, lightGBMBooster: LightGBMBooster): TrainedModel

    Gets the trained model given the train parameters and booster.

    Gets the trained model given the train parameters and booster.

    returns

    trained model.

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

    Gets the training parameters.

    Gets the training parameters.

    numTasks

    The total number of tasks.

    dataset

    The training dataset.

    numTasksPerExec

    The number of tasks per executor.

    returns

    train parameters.

    Attributes
    protected
  5. abstract def stringFromTrainedModel(model: TrainedModel): String
    Attributes
    protected
  6. abstract def transformSchema(schema: StructType): StructType
    Definition Classes
    PipelineStage
  7. abstract val uid: String
    Definition Classes
    BasicLogging

Concrete Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. final def $[T](param: Param[T]): T
    Attributes
    protected
    Definition Classes
    Params
  4. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  5. def afterTrainBatch(batchIndex: Int, dataset: Dataset[_], model: TrainedModel): Unit
  6. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  7. val baggingFraction: DoubleParam
    Definition Classes
    LightGBMFractionParams
  8. val baggingFreq: IntParam
    Definition Classes
    LightGBMParams
  9. val baggingSeed: IntParam
    Definition Classes
    LightGBMParams
  10. def beforeTrainBatch(batchIndex: Int, dataset: Dataset[_], model: Option[TrainedModel]): Unit
  11. val binSampleCount: IntParam
    Definition Classes
    LightGBMBinParams
  12. val boostFromAverage: BooleanParam
    Definition Classes
    LightGBMParams
  13. val boostingType: Param[String]
    Definition Classes
    LightGBMParams
  14. def castColumns(dataset: Dataset[_], trainingCols: Array[(String, Seq[DataType])]): DataFrame
    Attributes
    protected
  15. val categoricalSlotIndexes: IntArrayParam
    Definition Classes
    LightGBMSlotParams
  16. val categoricalSlotNames: StringArrayParam
    Definition Classes
    LightGBMSlotParams
  17. val chunkSize: IntParam
    Definition Classes
    LightGBMExecutionParams
  18. lazy val classNameHelper: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  19. final def clear(param: Param[_]): LightGBMBase.this.type
    Definition Classes
    Params
  20. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  21. def companionModelClassName: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  22. def copyValues[T <: Params](to: T, extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  23. lazy val copyrightLines: String
    Attributes
    protected
    Definition Classes
    BaseWrappable
  24. final def defaultCopy[T <: Params](extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  25. val defaultListenPort: IntParam
    Definition Classes
    LightGBMExecutionParams
  26. val delegate: Option[LightGBMDelegate]
    Definition Classes
    LightGBMParams
  27. val driverListenPort: IntParam
    Definition Classes
    LightGBMExecutionParams
  28. val dropRate: DoubleParam
    Definition Classes
    LightGBMDartParams
  29. val earlyStoppingRound: IntParam
    Definition Classes
    LightGBMLearnerParams
  30. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  31. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  32. def explainParam(param: Param[_]): String
    Definition Classes
    Params
  33. def explainParams(): String
    Definition Classes
    Params
  34. final def extractParamMap(): ParamMap
    Definition Classes
    Params
  35. final def extractParamMap(extra: ParamMap): ParamMap
    Definition Classes
    Params
  36. val featureFraction: DoubleParam
    Definition Classes
    LightGBMFractionParams
  37. final val featuresCol: Param[String]
    Definition Classes
    HasFeaturesCol
  38. val featuresShapCol: Param[String]
    Definition Classes
    LightGBMPredictionParams
  39. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  40. def fit(dataset: Dataset[_], paramMaps: Seq[ParamMap]): Seq[TrainedModel]
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  41. def fit(dataset: Dataset[_], paramMap: ParamMap): TrainedModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  42. def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): TrainedModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" ) @varargs()
  43. val fobj: FObjParam
    Definition Classes
    LightGBMObjectiveParams
  44. final def get[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  45. def getBaggingFraction: Double
    Definition Classes
    LightGBMFractionParams
  46. def getBaggingFreq: Int
    Definition Classes
    LightGBMParams
  47. def getBaggingSeed: Int
    Definition Classes
    LightGBMParams
  48. def getBinSampleCount: Int
    Definition Classes
    LightGBMBinParams
  49. def getBoostFromAverage: Boolean
    Definition Classes
    LightGBMParams
  50. def getBoostingType: String
    Definition Classes
    LightGBMParams
  51. def getCategoricalIndexes(featuresSchema: StructField): Array[Int]

    Retrieves the categorical indexes in the features column.

    Retrieves the categorical indexes in the features column.

    featuresSchema

    The schema of the features column

    returns

    the categorical indexes in the features column.

    Attributes
    protected
  52. def getCategoricalSlotIndexes: Array[Int]
    Definition Classes
    LightGBMSlotParams
  53. def getCategoricalSlotNames: Array[String]
    Definition Classes
    LightGBMSlotParams
  54. def getChunkSize: Int
    Definition Classes
    LightGBMExecutionParams
  55. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  56. def getColumnParams: ColumnParams
    Attributes
    protected
  57. def getDartParams: DartModeParams

    Constructs the DartModeParams

    Constructs the DartModeParams

    returns

    DartModeParams object containing parameters related to dart mode.

    Attributes
    protected
  58. def getDatasetParams(categoricalIndexes: Array[Int], numThreads: Int): String
  59. final def getDefault[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  60. def getDefaultListenPort: Int
    Definition Classes
    LightGBMExecutionParams
  61. def getDelegate: Option[LightGBMDelegate]
    Definition Classes
    LightGBMParams
  62. def getDriverListenPort: Int
    Definition Classes
    LightGBMExecutionParams
  63. def getDropRate: Double
    Definition Classes
    LightGBMDartParams
  64. def getEarlyStoppingRound: Int
    Definition Classes
    LightGBMLearnerParams
  65. def getExecutionParams(numTasksPerExec: Int): ExecutionParams

    Constructs the ExecutionParams.

    Constructs the ExecutionParams.

    returns

    ExecutionParams object containing parameters related to LightGBM execution.

    Attributes
    protected
  66. def getFObj: FObjTrait
    Definition Classes
    LightGBMObjectiveParams
  67. def getFeatureFraction: Double
    Definition Classes
    LightGBMFractionParams
  68. final def getFeaturesCol: String
    Definition Classes
    HasFeaturesCol
  69. def getFeaturesShapCol: String
    Definition Classes
    LightGBMPredictionParams
  70. def getImprovementTolerance: Double
    Definition Classes
    LightGBMLearnerParams
  71. def getInitScoreCol: String

    Definition Classes
    HasInitScoreCol
  72. def getIsProvideTrainingMetric: Boolean
    Definition Classes
    LightGBMParams
  73. final def getLabelCol: String
    Definition Classes
    HasLabelCol
  74. def getLambdaL1: Double
    Definition Classes
    LightGBMParams
  75. def getLambdaL2: Double
    Definition Classes
    LightGBMParams
  76. def getLeafPredictionCol: String
    Definition Classes
    LightGBMPredictionParams
  77. def getLearningRate: Double
    Definition Classes
    LightGBMParams
  78. def getMatrixType: String
    Definition Classes
    LightGBMExecutionParams
  79. def getMaxBin: Int
    Definition Classes
    LightGBMBinParams
  80. def getMaxBinByFeature: Array[Int]
    Definition Classes
    LightGBMParams
  81. def getMaxDeltaStep: Double
    Definition Classes
    LightGBMParams
  82. def getMaxDepth: Int
    Definition Classes
    LightGBMParams
  83. def getMaxDrop: Int
    Definition Classes
    LightGBMDartParams
  84. def getMetric: String
    Definition Classes
    LightGBMParams
  85. def getMinDataInLeaf: Int
    Definition Classes
    LightGBMParams
  86. def getMinGainToSplit: Double
    Definition Classes
    LightGBMParams
  87. def getMinSumHessianInLeaf: Double
    Definition Classes
    LightGBMParams
  88. def getModelString: String
    Definition Classes
    LightGBMParams
  89. def getNegBaggingFraction: Double
    Definition Classes
    LightGBMFractionParams
  90. def getNumBatches: Int
    Definition Classes
    LightGBMExecutionParams
  91. def getNumIterations: Int
    Definition Classes
    LightGBMParams
  92. def getNumLeaves: Int
    Definition Classes
    LightGBMParams
  93. def getNumTasks: Int
    Definition Classes
    LightGBMExecutionParams
  94. def getNumThreads: Int
    Definition Classes
    LightGBMExecutionParams
  95. def getObjective: String
    Definition Classes
    LightGBMObjectiveParams
  96. def getObjectiveParams: ObjectiveParams

    Constructs the ObjectiveParams.

    Constructs the ObjectiveParams.

    returns

    ObjectiveParams object containing parameters related to the objective function.

    Attributes
    protected
  97. def getOptGroupCol: Option[String]

    Optional group column for Ranking, set to None by default.

    Optional group column for Ranking, set to None by default.

    returns

    None

    Attributes
    protected
  98. final def getOrDefault[T](param: Param[T]): T
    Definition Classes
    Params
  99. def getParallelism: String
    Definition Classes
    LightGBMExecutionParams
  100. def getParam(paramName: String): Param[Any]
    Definition Classes
    Params
  101. def getPosBaggingFraction: Double
    Definition Classes
    LightGBMFractionParams
  102. def getPredictDisableShapeCheck: Boolean
    Definition Classes
    LightGBMPredictionParams
  103. def getRepartitionByGroupingColumn: Boolean
    Definition Classes
    LightGBMExecutionParams
  104. def getSkipDrop: Double
    Definition Classes
    LightGBMDartParams
  105. def getSlotNames: Array[String]
    Definition Classes
    LightGBMSlotParams
  106. def getSlotNamesWithMetadata(featuresSchema: StructField): Option[Array[String]]
  107. def getTimeout: Double
    Definition Classes
    LightGBMExecutionParams
  108. def getTopK: Int
    Definition Classes
    LightGBMExecutionParams
  109. def getTrainingCols: Array[(String, Seq[DataType])]
    Attributes
    protected
  110. def getUniformDrop: Boolean
    Definition Classes
    LightGBMDartParams
  111. def getUseBarrierExecutionMode: Boolean
    Definition Classes
    LightGBMExecutionParams
  112. def getUseSingleDatasetMode: Boolean
    Definition Classes
    LightGBMExecutionParams
  113. def getValidationIndicatorCol: String

    Definition Classes
    HasValidationIndicatorCol
  114. def getVerbosity: Int
    Definition Classes
    LightGBMParams
  115. def getWeightCol: String

    Definition Classes
    HasWeightCol
  116. def getXGBoostDartMode: Boolean
    Definition Classes
    LightGBMDartParams
  117. final def hasDefault[T](param: Param[T]): Boolean
    Definition Classes
    Params
  118. def hasParam(paramName: String): Boolean
    Definition Classes
    Params
  119. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  120. val improvementTolerance: DoubleParam
    Definition Classes
    LightGBMLearnerParams
  121. val initScoreCol: Param[String]

    The name of the initial score column

    The name of the initial score column

    Definition Classes
    HasInitScoreCol
  122. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  123. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  124. def innerTrain(dataset: Dataset[_], batchIndex: Int): TrainedModel

    Inner train method for LightGBM learners.

    Inner train method for LightGBM learners. Calculates the number of workers, creates a driver thread, and runs mapPartitions on the dataset.

    dataset

    The dataset to train on.

    batchIndex

    In running in batch training mode, gets the batch number.

    returns

    The LightGBM Model from the trained LightGBM Booster.

    Attributes
    protected
  125. final def isDefined(param: Param[_]): Boolean
    Definition Classes
    Params
  126. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  127. val isProvideTrainingMetric: BooleanParam
    Definition Classes
    LightGBMParams
  128. final def isSet(param: Param[_]): Boolean
    Definition Classes
    Params
  129. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  130. final val labelCol: Param[String]
    Definition Classes
    HasLabelCol
  131. val lambdaL1: DoubleParam
    Definition Classes
    LightGBMParams
  132. val lambdaL2: DoubleParam
    Definition Classes
    LightGBMParams
  133. val leafPredictionCol: Param[String]
    Definition Classes
    LightGBMPredictionParams
  134. val learningRate: DoubleParam
    Definition Classes
    LightGBMParams
  135. def log: Logger
    Attributes
    protected
    Definition Classes
    Logging
  136. def logBase(methodName: String): Unit
    Attributes
    protected
    Definition Classes
    BasicLogging
  137. def logClass(): Unit
    Definition Classes
    BasicLogging
  138. def logDebug(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  139. def logDebug(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  140. def logError(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  141. def logError(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  142. def logErrorBase(methodName: String, e: Exception): Unit
    Attributes
    protected
    Definition Classes
    BasicLogging
  143. def logFit[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  144. def logInfo(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  145. def logInfo(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  146. def logName: String
    Attributes
    protected
    Definition Classes
    Logging
  147. def logPredict[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  148. def logTrace(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  149. def logTrace(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  150. def logTrain[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  151. def logTransform[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  152. def logVerb[T](verb: String, f: ⇒ T): T
    Definition Classes
    BasicLogging
  153. def logWarning(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  154. def logWarning(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  155. def makePyFile(conf: CodegenConfig): Unit
    Definition Classes
    PythonWrappable
  156. def makeRFile(conf: CodegenConfig): Unit
    Definition Classes
    RWrappable
  157. val matrixType: Param[String]
    Definition Classes
    LightGBMExecutionParams
  158. val maxBin: IntParam
    Definition Classes
    LightGBMBinParams
  159. val maxBinByFeature: IntArrayParam
    Definition Classes
    LightGBMParams
  160. val maxDeltaStep: DoubleParam
    Definition Classes
    LightGBMParams
  161. val maxDepth: IntParam
    Definition Classes
    LightGBMParams
  162. val maxDrop: IntParam
    Definition Classes
    LightGBMDartParams
  163. val metric: Param[String]
    Definition Classes
    LightGBMParams
  164. val minDataInLeaf: IntParam
    Definition Classes
    LightGBMParams
  165. val minGainToSplit: DoubleParam
    Definition Classes
    LightGBMParams
  166. val minSumHessianInLeaf: DoubleParam
    Definition Classes
    LightGBMParams
  167. val modelString: Param[String]
    Definition Classes
    LightGBMParams
  168. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  169. val negBaggingFraction: DoubleParam
    Definition Classes
    LightGBMFractionParams
  170. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  171. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  172. val numBatches: IntParam
    Definition Classes
    LightGBMExecutionParams
  173. val numIterations: IntParam
    Definition Classes
    LightGBMParams
  174. val numLeaves: IntParam
    Definition Classes
    LightGBMParams
  175. val numTasks: IntParam
    Definition Classes
    LightGBMExecutionParams
  176. val numThreads: IntParam
    Definition Classes
    LightGBMExecutionParams
  177. val objective: Param[String]
    Definition Classes
    LightGBMObjectiveParams
  178. val parallelism: Param[String]
    Definition Classes
    LightGBMExecutionParams
  179. lazy val params: Array[Param[_]]
    Definition Classes
    Params
  180. val posBaggingFraction: DoubleParam
    Definition Classes
    LightGBMFractionParams
  181. val predictDisableShapeCheck: BooleanParam
    Definition Classes
    LightGBMPredictionParams
  182. def prepareDataframe(dataset: Dataset[_], numTasks: Int): DataFrame
    Attributes
    protected
  183. def preprocessData(dataset: DataFrame): DataFrame

    Allow algorithm specific preprocessing of dataset.

    Allow algorithm specific preprocessing of dataset.

    dataset

    The dataset to preprocess prior to training.

    returns

    The preprocessed data.

    Attributes
    protected
  184. def pyAdditionalMethods: String
    Definition Classes
    PythonWrappable
  185. lazy val pyClassDoc: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  186. lazy val pyClassName: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  187. def pyExtraEstimatorImports: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  188. def pyExtraEstimatorMethods: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  189. lazy val pyInheritedClasses: Seq[String]
    Attributes
    protected
    Definition Classes
    PythonWrappable
  190. def pyInitFunc(): String
    Definition Classes
    PythonWrappable
  191. lazy val pyInternalWrapper: Boolean
    Attributes
    protected
    Definition Classes
    PythonWrappable
  192. lazy val pyObjectBaseClass: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  193. def pyParamArg[T](p: Param[T]): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  194. def pyParamDefault[T](p: Param[T]): Option[String]
    Attributes
    protected
    Definition Classes
    PythonWrappable
  195. def pyParamGetter(p: Param[_]): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  196. def pyParamSetter(p: Param[_]): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  197. def pyParamsArgs: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  198. def pyParamsDefaults: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  199. lazy val pyParamsDefinitions: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  200. def pyParamsGetters: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  201. def pyParamsSetters: String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  202. def pythonClass(): String
    Attributes
    protected
    Definition Classes
    PythonWrappable
  203. def rClass(): String
    Attributes
    protected
    Definition Classes
    RWrappable
  204. def rDocString: String
    Attributes
    protected
    Definition Classes
    RWrappable
  205. def rExtraBodyLines: String
    Attributes
    protected
    Definition Classes
    RWrappable
  206. def rExtraInitLines: String
    Attributes
    protected
    Definition Classes
    RWrappable
  207. lazy val rFuncName: String
    Attributes
    protected
    Definition Classes
    RWrappable
  208. lazy val rInternalWrapper: Boolean
    Attributes
    protected
    Definition Classes
    RWrappable
  209. def rParamArg[T](p: Param[T]): String
    Attributes
    protected
    Definition Classes
    RWrappable
  210. def rParamsArgs: String
    Attributes
    protected
    Definition Classes
    RWrappable
  211. def rSetterLines: String
    Attributes
    protected
    Definition Classes
    RWrappable
  212. val repartitionByGroupingColumn: BooleanParam
    Definition Classes
    LightGBMExecutionParams
  213. def save(path: String): Unit
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )
  214. final def set(paramPair: ParamPair[_]): LightGBMBase.this.type
    Attributes
    protected
    Definition Classes
    Params
  215. final def set(param: String, value: Any): LightGBMBase.this.type
    Attributes
    protected
    Definition Classes
    Params
  216. final def set[T](param: Param[T], value: T): LightGBMBase.this.type
    Definition Classes
    Params
  217. def setBaggingFraction(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMFractionParams
  218. def setBaggingFreq(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  219. def setBaggingSeed(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  220. def setBinSampleCount(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMBinParams
  221. def setBoostFromAverage(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  222. def setBoostingType(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  223. def setCategoricalSlotIndexes(value: Array[Int]): LightGBMBase.this.type
    Definition Classes
    LightGBMSlotParams
  224. def setCategoricalSlotNames(value: Array[String]): LightGBMBase.this.type
    Definition Classes
    LightGBMSlotParams
  225. def setChunkSize(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  226. final def setDefault(paramPairs: ParamPair[_]*): LightGBMBase.this.type
    Attributes
    protected
    Definition Classes
    Params
  227. final def setDefault[T](param: Param[T], value: T): LightGBMBase.this.type
    Attributes
    protected
    Definition Classes
    Params
  228. def setDefaultListenPort(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  229. def setDelegate(delegate: LightGBMDelegate): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  230. def setDriverListenPort(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  231. def setDropRate(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMDartParams
  232. def setEarlyStoppingRound(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMLearnerParams
  233. def setFObj(value: FObjTrait): LightGBMBase.this.type
    Definition Classes
    LightGBMObjectiveParams
  234. def setFeatureFraction(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMFractionParams
  235. def setFeaturesShapCol(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMPredictionParams
  236. def setImprovementTolerance(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMLearnerParams
  237. def setInitScoreCol(value: String): LightGBMBase.this.type

    Definition Classes
    HasInitScoreCol
  238. def setIsProvideTrainingMetric(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  239. def setLambdaL1(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  240. def setLambdaL2(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  241. def setLeafPredictionCol(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMPredictionParams
  242. def setLearningRate(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  243. def setMatrixType(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  244. def setMaxBin(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMBinParams
  245. def setMaxBinByFeature(value: Array[Int]): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  246. def setMaxDeltaStep(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  247. def setMaxDepth(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  248. def setMaxDrop(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMDartParams
  249. def setMetric(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  250. def setMinDataInLeaf(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  251. def setMinGainToSplit(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  252. def setMinSumHessianInLeaf(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  253. def setModelString(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  254. def setNegBaggingFraction(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMFractionParams
  255. def setNumBatches(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  256. def setNumIterations(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  257. def setNumLeaves(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  258. def setNumTasks(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  259. def setNumThreads(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  260. def setObjective(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMObjectiveParams
  261. def setParallelism(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  262. def setPosBaggingFraction(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMFractionParams
  263. def setPredictDisableShapeCheck(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMPredictionParams
  264. def setRepartitionByGroupingColumn(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  265. def setSkipDrop(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMDartParams
  266. def setSlotNames(value: Array[String]): LightGBMBase.this.type
    Definition Classes
    LightGBMSlotParams
  267. def setTimeout(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  268. def setTopK(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  269. def setUniformDrop(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMDartParams
  270. def setUseBarrierExecutionMode(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  271. def setUseSingleDatasetMode(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  272. def setValidationIndicatorCol(value: String): LightGBMBase.this.type

    Definition Classes
    HasValidationIndicatorCol
  273. def setVerbosity(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  274. def setWeightCol(value: String): LightGBMBase.this.type

    Definition Classes
    HasWeightCol
  275. def setXGBoostDartMode(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMDartParams
  276. val skipDrop: DoubleParam
    Definition Classes
    LightGBMDartParams
  277. val slotNames: StringArrayParam
    Definition Classes
    LightGBMSlotParams
  278. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  279. val timeout: DoubleParam
    Definition Classes
    LightGBMExecutionParams
  280. def toString(): String
    Definition Classes
    Identifiable → AnyRef → Any
  281. val topK: IntParam
    Definition Classes
    LightGBMExecutionParams
  282. def train(dataset: Dataset[_]): TrainedModel

    Trains the LightGBM model.

    Trains the LightGBM model. If batches are specified, breaks training dataset into batches for training.

    dataset

    The input dataset to train.

    returns

    The trained model.

    Attributes
    protected
  283. def transformSchema(schema: StructType, logging: Boolean): StructType
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  284. val uniformDrop: BooleanParam
    Definition Classes
    LightGBMDartParams
  285. val useBarrierExecutionMode: BooleanParam
    Definition Classes
    LightGBMExecutionParams
  286. val useSingleDatasetMode: BooleanParam
    Definition Classes
    LightGBMExecutionParams
  287. val validationIndicatorCol: Param[String]

    The name of the validation indicator column

    The name of the validation indicator column

    Definition Classes
    HasValidationIndicatorCol
  288. val ver: String
    Definition Classes
    BasicLogging
  289. val verbosity: IntParam
    Definition Classes
    LightGBMParams
  290. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  291. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  292. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  293. val weightCol: Param[String]

    The name of the weight column

    The name of the weight column

    Definition Classes
    HasWeightCol
  294. def write: MLWriter
    Definition Classes
    DefaultParamsWritable → MLWritable
  295. val xgboostDartMode: BooleanParam
    Definition Classes
    LightGBMDartParams

Inherited from BasicLogging

Inherited from HasLabelCol

Inherited from HasFeaturesCol

Inherited from LightGBMParams

Inherited from LightGBMObjectiveParams

Inherited from LightGBMPredictionParams

Inherited from LightGBMDartParams

Inherited from LightGBMLearnerParams

Inherited from LightGBMBinParams

Inherited from LightGBMFractionParams

Inherited from LightGBMSlotParams

Inherited from LightGBMExecutionParams

Inherited from HasInitScoreCol

Inherited from HasValidationIndicatorCol

Inherited from HasWeightCol

Inherited from DefaultParamsWritable

Inherited from MLWritable

Inherited from Wrappable

Inherited from RWrappable

Inherited from PythonWrappable

Inherited from BaseWrappable

Inherited from Estimator[TrainedModel]

Inherited from PipelineStage

Inherited from Logging

Inherited from Params

Inherited from Serializable

Inherited from Serializable

Inherited from Identifiable

Inherited from AnyRef

Inherited from Any

getParam

param

setParam

Ungrouped