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 transformSchema(schema: StructType): StructType
    Definition Classes
    PipelineStage
  4. abstract val uid: String
    Definition Classes
    BasicLogging

Concrete Value Members

  1. def afterTrainBatch(batchIndex: Int, dataset: Dataset[_], model: TrainedModel): Unit
  2. val baggingFraction: DoubleParam
    Definition Classes
    LightGBMFractionParams
  3. val baggingFreq: IntParam
    Definition Classes
    LightGBMParams
  4. val baggingSeed: IntParam
    Definition Classes
    LightGBMParams
  5. def beforeTrainBatch(batchIndex: Int, dataset: Dataset[_], model: Option[TrainedModel]): Unit
  6. val binSampleCount: IntParam
    Definition Classes
    LightGBMBinParams
  7. val boostFromAverage: BooleanParam
    Definition Classes
    LightGBMParams
  8. val boostingType: Param[String]
    Definition Classes
    LightGBMParams
  9. val categoricalSlotIndexes: IntArrayParam
    Definition Classes
    LightGBMSlotParams
  10. val categoricalSlotNames: StringArrayParam
    Definition Classes
    LightGBMSlotParams
  11. val chunkSize: IntParam
    Definition Classes
    LightGBMExecutionParams
  12. final def clear(param: Param[_]): LightGBMBase.this.type
    Definition Classes
    Params
  13. val defaultListenPort: IntParam
    Definition Classes
    LightGBMExecutionParams
  14. val delegate: Option[LightGBMDelegate]
    Definition Classes
    LightGBMParams
  15. val driverListenPort: IntParam
    Definition Classes
    LightGBMExecutionParams
  16. val dropRate: DoubleParam
    Definition Classes
    LightGBMDartParams
  17. val earlyStoppingRound: IntParam
    Definition Classes
    LightGBMLearnerParams
  18. def explainParam(param: Param[_]): String
    Definition Classes
    Params
  19. def explainParams(): String
    Definition Classes
    Params
  20. final def extractParamMap(): ParamMap
    Definition Classes
    Params
  21. final def extractParamMap(extra: ParamMap): ParamMap
    Definition Classes
    Params
  22. val featureFraction: DoubleParam
    Definition Classes
    LightGBMFractionParams
  23. final val featuresCol: Param[String]
    Definition Classes
    HasFeaturesCol
  24. val featuresShapCol: Param[String]
    Definition Classes
    LightGBMPredictionParams
  25. def fit(dataset: Dataset[_], paramMaps: Seq[ParamMap]): Seq[TrainedModel]
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  26. def fit(dataset: Dataset[_], paramMap: ParamMap): TrainedModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  27. def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): TrainedModel
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" ) @varargs()
  28. val fobj: FObjParam
    Definition Classes
    LightGBMObjectiveParams
  29. final def get[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  30. def getBaggingFraction: Double
    Definition Classes
    LightGBMFractionParams
  31. def getBaggingFreq: Int
    Definition Classes
    LightGBMParams
  32. def getBaggingSeed: Int
    Definition Classes
    LightGBMParams
  33. def getBinSampleCount: Int
    Definition Classes
    LightGBMBinParams
  34. def getBoostFromAverage: Boolean
    Definition Classes
    LightGBMParams
  35. def getBoostingType: String
    Definition Classes
    LightGBMParams
  36. def getCategoricalSlotIndexes: Array[Int]
    Definition Classes
    LightGBMSlotParams
  37. def getCategoricalSlotNames: Array[String]
    Definition Classes
    LightGBMSlotParams
  38. def getChunkSize: Int
    Definition Classes
    LightGBMExecutionParams
  39. def getDatasetParams(categoricalIndexes: Array[Int], numThreads: Int): String
  40. final def getDefault[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  41. def getDefaultListenPort: Int
    Definition Classes
    LightGBMExecutionParams
  42. def getDelegate: Option[LightGBMDelegate]
    Definition Classes
    LightGBMParams
  43. def getDriverListenPort: Int
    Definition Classes
    LightGBMExecutionParams
  44. def getDropRate: Double
    Definition Classes
    LightGBMDartParams
  45. def getEarlyStoppingRound: Int
    Definition Classes
    LightGBMLearnerParams
  46. def getFObj: FObjTrait
    Definition Classes
    LightGBMObjectiveParams
  47. def getFeatureFraction: Double
    Definition Classes
    LightGBMFractionParams
  48. final def getFeaturesCol: String
    Definition Classes
    HasFeaturesCol
  49. def getFeaturesShapCol: String
    Definition Classes
    LightGBMPredictionParams
  50. def getImprovementTolerance: Double
    Definition Classes
    LightGBMLearnerParams
  51. def getInitScoreCol: String

    Definition Classes
    HasInitScoreCol
  52. def getIsProvideTrainingMetric: Boolean
    Definition Classes
    LightGBMParams
  53. final def getLabelCol: String
    Definition Classes
    HasLabelCol
  54. def getLambdaL1: Double
    Definition Classes
    LightGBMParams
  55. def getLambdaL2: Double
    Definition Classes
    LightGBMParams
  56. def getLeafPredictionCol: String
    Definition Classes
    LightGBMPredictionParams
  57. def getLearningRate: Double
    Definition Classes
    LightGBMParams
  58. def getMatrixType: String
    Definition Classes
    LightGBMExecutionParams
  59. def getMaxBin: Int
    Definition Classes
    LightGBMBinParams
  60. def getMaxBinByFeature: Array[Int]
    Definition Classes
    LightGBMParams
  61. def getMaxDeltaStep: Double
    Definition Classes
    LightGBMParams
  62. def getMaxDepth: Int
    Definition Classes
    LightGBMParams
  63. def getMaxDrop: Int
    Definition Classes
    LightGBMDartParams
  64. def getMetric: String
    Definition Classes
    LightGBMParams
  65. def getMinDataInLeaf: Int
    Definition Classes
    LightGBMParams
  66. def getMinGainToSplit: Double
    Definition Classes
    LightGBMParams
  67. def getMinSumHessianInLeaf: Double
    Definition Classes
    LightGBMParams
  68. def getModelString: String
    Definition Classes
    LightGBMParams
  69. def getNegBaggingFraction: Double
    Definition Classes
    LightGBMFractionParams
  70. def getNumBatches: Int
    Definition Classes
    LightGBMExecutionParams
  71. def getNumIterations: Int
    Definition Classes
    LightGBMParams
  72. def getNumLeaves: Int
    Definition Classes
    LightGBMParams
  73. def getNumTasks: Int
    Definition Classes
    LightGBMExecutionParams
  74. def getNumThreads: Int
    Definition Classes
    LightGBMExecutionParams
  75. def getObjective: String
    Definition Classes
    LightGBMObjectiveParams
  76. final def getOrDefault[T](param: Param[T]): T
    Definition Classes
    Params
  77. def getParallelism: String
    Definition Classes
    LightGBMExecutionParams
  78. def getParam(paramName: String): Param[Any]
    Definition Classes
    Params
  79. def getPosBaggingFraction: Double
    Definition Classes
    LightGBMFractionParams
  80. def getPredictDisableShapeCheck: Boolean
    Definition Classes
    LightGBMPredictionParams
  81. def getRepartitionByGroupingColumn: Boolean
    Definition Classes
    LightGBMExecutionParams
  82. def getSkipDrop: Double
    Definition Classes
    LightGBMDartParams
  83. def getSlotNames: Array[String]
    Definition Classes
    LightGBMSlotParams
  84. def getSlotNamesWithMetadata(featuresSchema: StructField): Option[Array[String]]
  85. def getTimeout: Double
    Definition Classes
    LightGBMExecutionParams
  86. def getTopK: Int
    Definition Classes
    LightGBMExecutionParams
  87. def getUniformDrop: Boolean
    Definition Classes
    LightGBMDartParams
  88. def getUseBarrierExecutionMode: Boolean
    Definition Classes
    LightGBMExecutionParams
  89. def getUseSingleDatasetMode: Boolean
    Definition Classes
    LightGBMExecutionParams
  90. def getValidationIndicatorCol: String

    Definition Classes
    HasValidationIndicatorCol
  91. def getVerbosity: Int
    Definition Classes
    LightGBMParams
  92. def getWeightCol: String

    Definition Classes
    HasWeightCol
  93. def getXGBoostDartMode: Boolean
    Definition Classes
    LightGBMDartParams
  94. final def hasDefault[T](param: Param[T]): Boolean
    Definition Classes
    Params
  95. def hasParam(paramName: String): Boolean
    Definition Classes
    Params
  96. val improvementTolerance: DoubleParam
    Definition Classes
    LightGBMLearnerParams
  97. val initScoreCol: Param[String]

    The name of the initial score column

    The name of the initial score column

    Definition Classes
    HasInitScoreCol
  98. final def isDefined(param: Param[_]): Boolean
    Definition Classes
    Params
  99. val isProvideTrainingMetric: BooleanParam
    Definition Classes
    LightGBMParams
  100. final def isSet(param: Param[_]): Boolean
    Definition Classes
    Params
  101. final val labelCol: Param[String]
    Definition Classes
    HasLabelCol
  102. val lambdaL1: DoubleParam
    Definition Classes
    LightGBMParams
  103. val lambdaL2: DoubleParam
    Definition Classes
    LightGBMParams
  104. val leafPredictionCol: Param[String]
    Definition Classes
    LightGBMPredictionParams
  105. val learningRate: DoubleParam
    Definition Classes
    LightGBMParams
  106. def logClass(): Unit
    Definition Classes
    BasicLogging
  107. def logFit[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  108. def logPredict[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  109. def logTrain[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  110. def logTransform[T](f: ⇒ T): T
    Definition Classes
    BasicLogging
  111. def logVerb[T](verb: String, f: ⇒ T): T
    Definition Classes
    BasicLogging
  112. def makePyFile(conf: CodegenConfig): Unit
    Definition Classes
    PythonWrappable
  113. def makeRFile(conf: CodegenConfig): Unit
    Definition Classes
    RWrappable
  114. val matrixType: Param[String]
    Definition Classes
    LightGBMExecutionParams
  115. val maxBin: IntParam
    Definition Classes
    LightGBMBinParams
  116. val maxBinByFeature: IntArrayParam
    Definition Classes
    LightGBMParams
  117. val maxDeltaStep: DoubleParam
    Definition Classes
    LightGBMParams
  118. val maxDepth: IntParam
    Definition Classes
    LightGBMParams
  119. val maxDrop: IntParam
    Definition Classes
    LightGBMDartParams
  120. val metric: Param[String]
    Definition Classes
    LightGBMParams
  121. val minDataInLeaf: IntParam
    Definition Classes
    LightGBMParams
  122. val minGainToSplit: DoubleParam
    Definition Classes
    LightGBMParams
  123. val minSumHessianInLeaf: DoubleParam
    Definition Classes
    LightGBMParams
  124. val modelString: Param[String]
    Definition Classes
    LightGBMParams
  125. val negBaggingFraction: DoubleParam
    Definition Classes
    LightGBMFractionParams
  126. val numBatches: IntParam
    Definition Classes
    LightGBMExecutionParams
  127. val numIterations: IntParam
    Definition Classes
    LightGBMParams
  128. val numLeaves: IntParam
    Definition Classes
    LightGBMParams
  129. val numTasks: IntParam
    Definition Classes
    LightGBMExecutionParams
  130. val numThreads: IntParam
    Definition Classes
    LightGBMExecutionParams
  131. val objective: Param[String]
    Definition Classes
    LightGBMObjectiveParams
  132. val parallelism: Param[String]
    Definition Classes
    LightGBMExecutionParams
  133. lazy val params: Array[Param[_]]
    Definition Classes
    Params
  134. val posBaggingFraction: DoubleParam
    Definition Classes
    LightGBMFractionParams
  135. val predictDisableShapeCheck: BooleanParam
    Definition Classes
    LightGBMPredictionParams
  136. def pyAdditionalMethods: String
    Definition Classes
    PythonWrappable
  137. def pyInitFunc(): String
    Definition Classes
    PythonWrappable
  138. val repartitionByGroupingColumn: BooleanParam
    Definition Classes
    LightGBMExecutionParams
  139. def save(path: String): Unit
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )
  140. final def set[T](param: Param[T], value: T): LightGBMBase.this.type
    Definition Classes
    Params
  141. def setBaggingFraction(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMFractionParams
  142. def setBaggingFreq(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  143. def setBaggingSeed(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  144. def setBinSampleCount(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMBinParams
  145. def setBoostFromAverage(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  146. def setBoostingType(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  147. def setCategoricalSlotIndexes(value: Array[Int]): LightGBMBase.this.type
    Definition Classes
    LightGBMSlotParams
  148. def setCategoricalSlotNames(value: Array[String]): LightGBMBase.this.type
    Definition Classes
    LightGBMSlotParams
  149. def setChunkSize(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  150. def setDefaultListenPort(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  151. def setDelegate(delegate: LightGBMDelegate): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  152. def setDriverListenPort(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  153. def setDropRate(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMDartParams
  154. def setEarlyStoppingRound(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMLearnerParams
  155. def setFObj(value: FObjTrait): LightGBMBase.this.type
    Definition Classes
    LightGBMObjectiveParams
  156. def setFeatureFraction(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMFractionParams
  157. def setFeaturesShapCol(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMPredictionParams
  158. def setImprovementTolerance(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMLearnerParams
  159. def setInitScoreCol(value: String): LightGBMBase.this.type

    Definition Classes
    HasInitScoreCol
  160. def setIsProvideTrainingMetric(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  161. def setLambdaL1(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  162. def setLambdaL2(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  163. def setLeafPredictionCol(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMPredictionParams
  164. def setLearningRate(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  165. def setMatrixType(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  166. def setMaxBin(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMBinParams
  167. def setMaxBinByFeature(value: Array[Int]): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  168. def setMaxDeltaStep(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  169. def setMaxDepth(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  170. def setMaxDrop(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMDartParams
  171. def setMetric(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  172. def setMinDataInLeaf(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  173. def setMinGainToSplit(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  174. def setMinSumHessianInLeaf(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  175. def setModelString(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  176. def setNegBaggingFraction(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMFractionParams
  177. def setNumBatches(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  178. def setNumIterations(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  179. def setNumLeaves(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMParams
  180. def setNumTasks(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  181. def setNumThreads(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  182. def setObjective(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMObjectiveParams
  183. def setParallelism(value: String): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  184. def setPosBaggingFraction(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMFractionParams
  185. def setPredictDisableShapeCheck(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMPredictionParams
  186. def setRepartitionByGroupingColumn(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  187. def setSkipDrop(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMDartParams
  188. def setSlotNames(value: Array[String]): LightGBMBase.this.type
    Definition Classes
    LightGBMSlotParams
  189. def setTimeout(value: Double): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  190. def setTopK(value: Int): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  191. def setUniformDrop(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMDartParams
  192. def setUseBarrierExecutionMode(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  193. def setUseSingleDatasetMode(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMExecutionParams
  194. def setValidationIndicatorCol(value: String): LightGBMBase.this.type

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

    Definition Classes
    HasWeightCol
  197. def setXGBoostDartMode(value: Boolean): LightGBMBase.this.type
    Definition Classes
    LightGBMDartParams
  198. val skipDrop: DoubleParam
    Definition Classes
    LightGBMDartParams
  199. val slotNames: StringArrayParam
    Definition Classes
    LightGBMSlotParams
  200. val timeout: DoubleParam
    Definition Classes
    LightGBMExecutionParams
  201. def toString(): String
    Definition Classes
    Identifiable → AnyRef → Any
  202. val topK: IntParam
    Definition Classes
    LightGBMExecutionParams
  203. val uniformDrop: BooleanParam
    Definition Classes
    LightGBMDartParams
  204. val useBarrierExecutionMode: BooleanParam
    Definition Classes
    LightGBMExecutionParams
  205. val useSingleDatasetMode: BooleanParam
    Definition Classes
    LightGBMExecutionParams
  206. val validationIndicatorCol: Param[String]

    The name of the validation indicator column

    The name of the validation indicator column

    Definition Classes
    HasValidationIndicatorCol
  207. val ver: String
    Definition Classes
    BasicLogging
  208. val verbosity: IntParam
    Definition Classes
    LightGBMParams
  209. val weightCol: Param[String]

    The name of the weight column

    The name of the weight column

    Definition Classes
    HasWeightCol
  210. def write: MLWriter
    Definition Classes
    DefaultParamsWritable → MLWritable
  211. val xgboostDartMode: BooleanParam
    Definition Classes
    LightGBMDartParams