LightGBMRegressor

Instance Constructors

new LightGBMRegressor()
new LightGBMRegressor(uid: String)

uid
The unique ID.

Value Members

final def !=(arg0: Any): Boolean

Definition Classes
AnyRef → Any
final def ##(): Int

Definition Classes
AnyRef → Any
final def $[T](param: Param[T]): T

Attributes
protected
Definition Classes
Params
final def ==(arg0: Any): Boolean

Definition Classes
AnyRef → Any
val alpha: DoubleParam
final def asInstanceOf[T0]: T0

Definition Classes
Any
val baggingFraction: DoubleParam

Definition Classes
LightGBMParams
val baggingFreq: IntParam

Definition Classes
LightGBMParams
val baggingSeed: IntParam

Definition Classes
LightGBMParams
val boostFromAverage: BooleanParam

Definition Classes
LightGBMParams
val boostingType: Param[String]

Definition Classes
LightGBMParams
val categoricalSlotIndexes: IntArrayParam

Definition Classes
LightGBMParams
val categoricalSlotNames: StringArrayParam

Definition Classes
LightGBMParams
final def clear(param: Param[_]): LightGBMRegressor.this.type

Definition Classes
Params
def clone(): AnyRef

Attributes
protected[java.lang]
Definition Classes
AnyRef
Annotations
@throws( ... )
def copy(extra: ParamMap): LightGBMRegressor

Definition Classes
LightGBMRegressor → Predictor → Estimator → PipelineStage → Params
def copyValues[T <: Params](to: T, extra: ParamMap): T

Attributes
protected
Definition Classes
Params
final def defaultCopy[T <: Params](extra: ParamMap): T

Attributes
protected
Definition Classes
Params
val defaultListenPort: IntParam

Definition Classes
LightGBMParams
val earlyStoppingRound: IntParam

Definition Classes
LightGBMParams
final def eq(arg0: AnyRef): Boolean

Definition Classes
AnyRef
def equals(arg0: Any): Boolean

Definition Classes
AnyRef → Any
def explainParam(param: Param[_]): String

Definition Classes
Params
def explainParams(): String

Definition Classes
Params
def extractLabeledPoints(dataset: Dataset[_]): RDD[LabeledPoint]

Attributes
protected
Definition Classes
Predictor
final def extractParamMap(): ParamMap

Definition Classes
Params
final def extractParamMap(extra: ParamMap): ParamMap

Definition Classes
Params
val featureFraction: DoubleParam

Definition Classes
LightGBMParams
final val featuresCol: Param[String]

Definition Classes
HasFeaturesCol
def finalize(): Unit

Attributes
protected[java.lang]
Definition Classes
AnyRef
Annotations
@throws( classOf[java.lang.Throwable] )
def fit(dataset: Dataset[_]): LightGBMRegressionModel

Definition Classes
Predictor → Estimator
def fit(dataset: Dataset[_], paramMaps: Array[ParamMap]): Seq[LightGBMRegressionModel]

Definition Classes
Estimator
Annotations
@Since( "2.0.0" )
def fit(dataset: Dataset[_], paramMap: ParamMap): LightGBMRegressionModel

Definition Classes
Estimator
Annotations
@Since( "2.0.0" )
def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): LightGBMRegressionModel

Definition Classes
Estimator
Annotations
@Since( "2.0.0" ) @varargs()
final def get[T](param: Param[T]): Option[T]

Definition Classes
Params
def getAlpha: Double
def getBaggingFraction: Double

Definition Classes
LightGBMParams
def getBaggingFreq: Int

Definition Classes
LightGBMParams
def getBaggingSeed: Int

Definition Classes
LightGBMParams
def getBoostFromAverage: Boolean

Definition Classes
LightGBMParams
def getBoostingType: String

Definition Classes
LightGBMParams
def getCategoricalSlotIndexes: Array[Int]

Definition Classes
LightGBMParams
def getCategoricalSlotNames: Array[String]

Definition Classes
LightGBMParams
final def getClass(): Class[_]

Definition Classes
AnyRef → Any
final def getDefault[T](param: Param[T]): Option[T]

Definition Classes
Params
def getDefaultListenPort: Int

Definition Classes
LightGBMParams
def getEarlyStoppingRound: Int

Definition Classes
LightGBMParams
def getFeatureFraction: Double

Definition Classes
LightGBMParams
final def getFeaturesCol: String

Definition Classes
HasFeaturesCol
final def getLabelCol: String

Definition Classes
HasLabelCol
def getLambdaL1: Double

Definition Classes
LightGBMParams
def getLambdaL2: Double

Definition Classes
LightGBMParams
def getLearningRate: Double

Definition Classes
LightGBMParams
def getMaxBin: Int

Definition Classes
LightGBMParams
def getMaxDepth: Int

Definition Classes
LightGBMParams
def getMinSumHessianInLeaf: Double

Definition Classes
LightGBMParams
def getModel(trainParams: TrainParams, lightGBMBooster: LightGBMBooster): LightGBMRegressionModel

Definition Classes
LightGBMRegressor → LightGBMBase
def getModelString: String

Definition Classes
LightGBMParams
def getNumIterations: Int

Definition Classes
LightGBMParams
def getNumLeaves: Int

Definition Classes
LightGBMParams
def getObjective: String

Definition Classes
LightGBMParams
final def getOrDefault[T](param: Param[T]): T

Definition Classes
Params
def getParallelism: String

Definition Classes
LightGBMParams
def getParam(paramName: String): Param[Any]

Definition Classes
Params
final def getPredictionCol: String

Definition Classes
HasPredictionCol
def getTimeout: Double

Definition Classes
LightGBMParams
def getTrainParams(numWorkers: Int, categoricalIndexes: Array[Int], dataset: Dataset[_]): TrainParams

Definition Classes
LightGBMRegressor → LightGBMBase
def getTweedieVariancePower: Double
def getValidationIndicatorCol: String

Definition Classes
HasValidationIndicatorCol
def getVerbosity: Int

Definition Classes
LightGBMParams
def getWeightCol: String

Definition Classes
HasWeightCol
final def hasDefault[T](param: Param[T]): Boolean

Definition Classes
Params
def hasParam(paramName: String): Boolean

Definition Classes
Params
def hashCode(): Int

Definition Classes
AnyRef → Any
def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean

Attributes
protected
Definition Classes
Logging
def initializeLogIfNecessary(isInterpreter: Boolean): Unit

Attributes
protected
Definition Classes
Logging
final def isDefined(param: Param[_]): Boolean

Definition Classes
Params
final def isInstanceOf[T0]: Boolean

Definition Classes
Any
final def isSet(param: Param[_]): Boolean

Definition Classes
Params
def isTraceEnabled(): Boolean

Attributes
protected
Definition Classes
Logging
final val labelCol: Param[String]

Definition Classes
HasLabelCol
val lambdaL1: DoubleParam

Definition Classes
LightGBMParams
val lambdaL2: DoubleParam

Definition Classes
LightGBMParams
val learningRate: DoubleParam

Definition Classes
LightGBMParams
def log: Logger

Attributes
protected
Definition Classes
Logging
def logDebug(msg: ⇒ String, throwable: Throwable): Unit

Attributes
protected
Definition Classes
Logging
def logDebug(msg: ⇒ String): Unit

Attributes
protected
Definition Classes
Logging
def logError(msg: ⇒ String, throwable: Throwable): Unit

Attributes
protected
Definition Classes
Logging
def logError(msg: ⇒ String): Unit

Attributes
protected
Definition Classes
Logging
def logInfo(msg: ⇒ String, throwable: Throwable): Unit

Attributes
protected
Definition Classes
Logging
def logInfo(msg: ⇒ String): Unit

Attributes
protected
Definition Classes
Logging
def logName: String

Attributes
protected
Definition Classes
Logging
def logTrace(msg: ⇒ String, throwable: Throwable): Unit

Attributes
protected
Definition Classes
Logging
def logTrace(msg: ⇒ String): Unit

Attributes
protected
Definition Classes
Logging
def logWarning(msg: ⇒ String, throwable: Throwable): Unit

Attributes
protected
Definition Classes
Logging
def logWarning(msg: ⇒ String): Unit

Attributes
protected
Definition Classes
Logging
val maxBin: IntParam

Definition Classes
LightGBMParams
val maxDepth: IntParam

Definition Classes
LightGBMParams
val minSumHessianInLeaf: DoubleParam

Definition Classes
LightGBMParams
val modelString: Param[String]

Definition Classes
LightGBMParams
final def ne(arg0: AnyRef): Boolean

Definition Classes
AnyRef
final def notify(): Unit

Definition Classes
AnyRef
final def notifyAll(): Unit

Definition Classes
AnyRef
val numIterations: IntParam

Definition Classes
LightGBMParams
val numLeaves: IntParam

Definition Classes
LightGBMParams
val objective: Param[String]

Definition Classes
LightGBMParams
val parallelism: Param[String]

Definition Classes
LightGBMParams
lazy val params: Array[Param[_]]

Definition Classes
Params
final val predictionCol: Param[String]

Definition Classes
HasPredictionCol
def save(path: String): Unit

Definition Classes
MLWritable
Annotations
@Since( "1.6.0" ) @throws( ... )
final def set(paramPair: ParamPair[_]): LightGBMRegressor.this.type

Attributes
protected
Definition Classes
Params
final def set(param: String, value: Any): LightGBMRegressor.this.type

Attributes
protected
Definition Classes
Params
final def set[T](param: Param[T], value: T): LightGBMRegressor.this.type

Definition Classes
Params
def setAlpha(value: Double): LightGBMRegressor.this.type
def setBaggingFraction(value: Double): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setBaggingFreq(value: Int): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setBaggingSeed(value: Int): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setBoostFromAverage(value: Boolean): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setBoostingType(value: String): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setCategoricalSlotIndexes(value: Array[Int]): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setCategoricalSlotNames(value: Array[String]): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
final def setDefault(paramPairs: ParamPair[_]*): LightGBMRegressor.this.type

Attributes
protected
Definition Classes
Params
final def setDefault[T](param: Param[T], value: T): LightGBMRegressor.this.type

Attributes
protected
Definition Classes
Params
def setDefaultListenPort(value: Int): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setEarlyStoppingRound(value: Int): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setFeatureFraction(value: Double): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setFeaturesCol(value: String): LightGBMRegressor

Definition Classes
Predictor
def setLabelCol(value: String): LightGBMRegressor

Definition Classes
Predictor
def setLambdaL1(value: Double): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setLambdaL2(value: Double): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setLearningRate(value: Double): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setMaxBin(value: Int): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setMaxDepth(value: Int): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setMinSumHessianInLeaf(value: Double): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setModelString(value: String): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setNumIterations(value: Int): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setNumLeaves(value: Int): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setObjective(value: String): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setParallelism(value: String): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setPredictionCol(value: String): LightGBMRegressor

Definition Classes
Predictor
def setTimeout(value: Double): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setTweedieVariancePower(value: Double): LightGBMRegressor.this.type
def setValidationIndicatorCol(value: String): LightGBMRegressor.this.type

Definition Classes
HasValidationIndicatorCol
def setVerbosity(value: Int): LightGBMRegressor.this.type

Definition Classes
LightGBMParams
def setWeightCol(value: String): LightGBMRegressor.this.type

Definition Classes
HasWeightCol
final def synchronized[T0](arg0: ⇒ T0): T0

Definition Classes
AnyRef
val timeout: DoubleParam

Definition Classes
LightGBMParams
def toString(): String

Definition Classes
Identifiable → AnyRef → Any
def train(dataset: Dataset[_]): LightGBMRegressionModel

Trains the LightGBM model.
Trains the LightGBM model.
dataset
The input dataset to train.
returns
The trained model.

Attributes
protected
Definition Classes
LightGBMBase
def transformSchema(schema: StructType): StructType

Definition Classes
Predictor → PipelineStage
def transformSchema(schema: StructType, logging: Boolean): StructType

Attributes
protected
Definition Classes
PipelineStage
Annotations
@DeveloperApi()
val tweedieVariancePower: DoubleParam
val uid: String

The unique ID.
The unique ID.

Definition Classes
LightGBMRegressor → Identifiable
def validateAndTransformSchema(schema: StructType, fitting: Boolean, featuresDataType: DataType): StructType

Attributes
protected
Definition Classes
PredictorParams
val validationIndicatorCol: Param[String]

The name of the validation indicator column
The name of the validation indicator column

Definition Classes
HasValidationIndicatorCol
val verbosity: IntParam

Definition Classes
LightGBMParams
final def wait(): Unit

Definition Classes
AnyRef
Annotations
@throws( ... )
final def wait(arg0: Long, arg1: Int): Unit

Definition Classes
AnyRef
Annotations
@throws( ... )
final def wait(arg0: Long): Unit

Definition Classes
AnyRef
Annotations
@throws( ... )
val weightCol: Param[String]

The name of the weight column
The name of the weight column

Definition Classes
HasWeightCol
def write: MLWriter

Definition Classes
DefaultParamsWritable → MLWritable

Inherited from LightGBMBase[LightGBMRegressionModel]

Inherited from LightGBMParams

Inherited from HasValidationIndicatorCol

Inherited from HasWeightCol

Inherited from DefaultParamsWritable

Inherited from MLWritable

Inherited from Wrappable

Inherited from BaseRegressor[Vector, LightGBMRegressor, LightGBMRegressionModel]

Inherited from Regressor[Vector, LightGBMRegressor, LightGBMRegressionModel]

Inherited from Predictor[Vector, LightGBMRegressor, LightGBMRegressionModel]

Inherited from PredictorParams

Inherited from HasPredictionCol

Inherited from org.apache.spark.ml.param.shared.HasFeaturesCol

Inherited from org.apache.spark.ml.param.shared.HasLabelCol

Inherited from Estimator[LightGBMRegressionModel]

Inherited from PipelineStage

Inherited from org.apache.spark.internal.Logging

Inherited from Params

Inherited from Serializable

Inherited from Identifiable

Inherited from AnyRef

Inherited from Any

Parameters

A list of parameter keys this algorithm can take. Users can set and get the parameter values through setters and getters

Parameter setters

Parameter getters

Members

Related Docs: object LightGBMRegressor | package spark

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

Instance Constructors

new LightGBMRegressor()

new LightGBMRegressor(uid: String)

Value Members

final def !=(arg0: Any): Boolean

final def ##(): Int

final def $[T](param: Param[T]): T

final def ==(arg0: Any): Boolean

val alpha: DoubleParam

final def asInstanceOf[T0]: T0

val baggingFraction: DoubleParam

val baggingFreq: IntParam

val baggingSeed: IntParam

val boostFromAverage: BooleanParam

val boostingType: Param[String]

val categoricalSlotIndexes: IntArrayParam

val categoricalSlotNames: StringArrayParam

final def clear(param: Param[_]): LightGBMRegressor.this.type

def clone(): AnyRef

def copy(extra: ParamMap): LightGBMRegressor

def copyValues[T <: Params](to: T, extra: ParamMap): T

final def defaultCopy[T <: Params](extra: ParamMap): T

val defaultListenPort: IntParam

val earlyStoppingRound: IntParam

final def eq(arg0: AnyRef): Boolean

def equals(arg0: Any): Boolean

def explainParam(param: Param[_]): String

def explainParams(): String

def extractLabeledPoints(dataset: Dataset[_]): RDD[LabeledPoint]

final def extractParamMap(): ParamMap

final def extractParamMap(extra: ParamMap): ParamMap

val featureFraction: DoubleParam

final val featuresCol: Param[String]

def finalize(): Unit

def fit(dataset: Dataset[_]): LightGBMRegressionModel

def fit(dataset: Dataset[_], paramMaps: Array[ParamMap]): Seq[LightGBMRegressionModel]

def fit(dataset: Dataset[_], paramMap: ParamMap): LightGBMRegressionModel

def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): LightGBMRegressionModel

final def get[T](param: Param[T]): Option[T]

def getAlpha: Double

def getBaggingFraction: Double

def getBaggingFreq: Int

def getBaggingSeed: Int

def getBoostFromAverage: Boolean

def getBoostingType: String

def getCategoricalSlotIndexes: Array[Int]

def getCategoricalSlotNames: Array[String]

final def getClass(): Class[_]

final def getDefault[T](param: Param[T]): Option[T]

def getDefaultListenPort: Int

def getEarlyStoppingRound: Int

def getFeatureFraction: Double

final def getFeaturesCol: String

final def getLabelCol: String

def getLambdaL1: Double

def getLambdaL2: Double

def getLearningRate: Double

def getMaxBin: Int

def getMaxDepth: Int

def getMinSumHessianInLeaf: Double

def getModel(trainParams: TrainParams, lightGBMBooster: LightGBMBooster): LightGBMRegressionModel

def getModelString: String

def getNumIterations: Int

def getNumLeaves: Int

def getObjective: String

final def getOrDefault[T](param: Param[T]): T

def getParallelism: String

def getParam(paramName: String): Param[Any]

final def getPredictionCol: String

def getTimeout: Double

def getTrainParams(numWorkers: Int, categoricalIndexes: Array[Int], dataset: Dataset[_]): TrainParams

def getTweedieVariancePower: Double

def getValidationIndicatorCol: String

def getVerbosity: Int

def getWeightCol: String

final def hasDefault[T](param: Param[T]): Boolean

def hasParam(paramName: String): Boolean