class LightGBMRanker extends Ranker[Vector, LightGBMRanker, LightGBMRankerModel] with LightGBMBase[LightGBMRankerModel] with SynapseMLLogging
Trains a LightGBMRanker 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
- Alphabetic
- By Inheritance
- LightGBMRanker
- LightGBMBase
- SynapseMLLogging
- LightGBMPerformance
- ComplexParamsWritable
- LightGBMParams
- LightGBMCategoricalParams
- LightGBMSeedParams
- LightGBMObjectiveParams
- LightGBMPredictionParams
- LightGBMDartParams
- LightGBMDatasetParams
- LightGBMLearnerParams
- LightGBMBinParams
- LightGBMFractionParams
- LightGBMSlotParams
- LightGBMExecutionParams
- HasInitScoreCol
- HasValidationIndicatorCol
- HasWeightCol
- DefaultParamsWritable
- MLWritable
- Wrappable
- DotnetWrappable
- RWrappable
- PythonWrappable
- BaseWrappable
- Ranker
- HasGroupCol
- Predictor
- PredictorParams
- HasPredictionCol
- HasFeaturesCol
- HasLabelCol
- Estimator
- PipelineStage
- Logging
- Params
- Serializable
- Serializable
- Identifiable
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Value Members
-
def
afterTrainBatch(batchIndex: Int, dataset: Dataset[_], model: LightGBMRankerModel): Unit
- Definition Classes
- LightGBMBase
-
val
baggingFraction: DoubleParam
- Definition Classes
- LightGBMFractionParams
-
val
baggingFreq: IntParam
- Definition Classes
- LightGBMParams
-
val
baggingSeed: IntParam
- Definition Classes
- LightGBMSeedParams
-
def
beforeTrainBatch(batchIndex: Int, dataset: Dataset[_], model: Option[LightGBMRankerModel]): Unit
- Definition Classes
- LightGBMBase
-
val
binSampleCount: IntParam
- Definition Classes
- LightGBMBinParams
-
val
boostFromAverage: BooleanParam
- Definition Classes
- LightGBMParams
-
val
boostingType: Param[String]
- Definition Classes
- LightGBMParams
-
val
catSmooth: DoubleParam
- Definition Classes
- LightGBMCategoricalParams
-
val
categoricalSlotIndexes: IntArrayParam
- Definition Classes
- LightGBMSlotParams
-
val
categoricalSlotNames: StringArrayParam
- Definition Classes
- LightGBMSlotParams
-
val
catl2: DoubleParam
- Definition Classes
- LightGBMCategoricalParams
-
val
chunkSize: IntParam
- Definition Classes
- LightGBMExecutionParams
-
final
def
clear(param: Param[_]): LightGBMRanker.this.type
- Definition Classes
- Params
-
def
copy(extra: ParamMap): LightGBMRanker
- Definition Classes
- LightGBMRanker → Predictor → Estimator → PipelineStage → Params
-
val
dataRandomSeed: IntParam
- Definition Classes
- LightGBMSeedParams
-
val
dataTransferMode: Param[String]
- Definition Classes
- LightGBMExecutionParams
-
val
defaultListenPort: IntParam
- Definition Classes
- LightGBMExecutionParams
-
val
delegate: Option[LightGBMDelegate]
- Definition Classes
- LightGBMParams
-
val
deterministic: BooleanParam
- Definition Classes
- LightGBMSeedParams
-
def
dotnetAdditionalMethods: String
- Definition Classes
- DotnetWrappable
-
val
driverListenPort: IntParam
- Definition Classes
- LightGBMExecutionParams
-
val
dropRate: DoubleParam
- Definition Classes
- LightGBMDartParams
-
val
dropSeed: IntParam
- Definition Classes
- LightGBMSeedParams
-
val
earlyStoppingRound: IntParam
- Definition Classes
- LightGBMLearnerParams
- val evalAt: IntArrayParam
-
def
explainParam(param: Param[_]): String
- Definition Classes
- Params
-
def
explainParams(): String
- Definition Classes
- Params
-
val
extraSeed: IntParam
- Definition Classes
- LightGBMSeedParams
-
final
def
extractParamMap(): ParamMap
- Definition Classes
- Params
-
final
def
extractParamMap(extra: ParamMap): ParamMap
- Definition Classes
- Params
-
val
featureFraction: DoubleParam
- Definition Classes
- LightGBMFractionParams
-
val
featureFractionByNode: DoubleParam
- Definition Classes
- LightGBMFractionParams
-
val
featureFractionSeed: IntParam
- Definition Classes
- LightGBMSeedParams
-
final
val
featuresCol: Param[String]
- Definition Classes
- HasFeaturesCol
-
val
featuresShapCol: Param[String]
- Definition Classes
- LightGBMPredictionParams
-
def
fit(dataset: Dataset[_]): LightGBMRankerModel
- Definition Classes
- Predictor → Estimator
-
def
fit(dataset: Dataset[_], paramMaps: Seq[ParamMap]): Seq[LightGBMRankerModel]
- Definition Classes
- Estimator
- Annotations
- @Since( "2.0.0" )
-
def
fit(dataset: Dataset[_], paramMap: ParamMap): LightGBMRankerModel
- Definition Classes
- Estimator
- Annotations
- @Since( "2.0.0" )
-
def
fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): LightGBMRankerModel
- Definition Classes
- Estimator
- Annotations
- @Since( "2.0.0" ) @varargs()
-
val
fobj: FObjParam
- Definition Classes
- LightGBMObjectiveParams
-
final
def
get[T](param: Param[T]): Option[T]
- Definition Classes
- Params
-
def
getAllPerformanceMeasures: Option[Array[InstrumentationMeasures]]
- Definition Classes
- LightGBMPerformance
-
def
getBaggingFraction: Double
- Definition Classes
- LightGBMFractionParams
-
def
getBaggingFreq: Int
- Definition Classes
- LightGBMParams
-
def
getBaggingSeed: Int
- Definition Classes
- LightGBMSeedParams
-
def
getBinSampleCount: Int
- Definition Classes
- LightGBMBinParams
-
def
getBoostFromAverage: Boolean
- Definition Classes
- LightGBMParams
-
def
getBoostingType: String
- Definition Classes
- LightGBMParams
-
def
getCatSmooth: Double
- Definition Classes
- LightGBMCategoricalParams
-
def
getCategoricalSlotIndexes: Array[Int]
- Definition Classes
- LightGBMSlotParams
-
def
getCategoricalSlotNames: Array[String]
- Definition Classes
- LightGBMSlotParams
-
def
getCatl2: Double
- Definition Classes
- LightGBMCategoricalParams
-
def
getChunkSize: Int
- Definition Classes
- LightGBMExecutionParams
-
def
getDataRandomSeed: Int
- Definition Classes
- LightGBMSeedParams
-
def
getDataTransferMode: String
- Definition Classes
- LightGBMExecutionParams
-
final
def
getDefault[T](param: Param[T]): Option[T]
- Definition Classes
- Params
-
def
getDefaultListenPort: Int
- Definition Classes
- LightGBMExecutionParams
-
def
getDelegate: Option[LightGBMDelegate]
- Definition Classes
- LightGBMParams
-
def
getDeterministic: Boolean
- Definition Classes
- LightGBMSeedParams
-
def
getDriverListenPort: Int
- Definition Classes
- LightGBMExecutionParams
-
def
getDropRate: Double
- Definition Classes
- LightGBMDartParams
-
def
getDropSeed: Int
- Definition Classes
- LightGBMSeedParams
-
def
getEarlyStoppingRound: Int
- Definition Classes
- LightGBMLearnerParams
- def getEvalAt: Array[Int]
-
def
getExtraSeed: Int
- Definition Classes
- LightGBMSeedParams
-
def
getFObj: FObjTrait
- Definition Classes
- LightGBMObjectiveParams
-
def
getFeatureFraction: Double
- Definition Classes
- LightGBMFractionParams
-
def
getFeatureFractionByNode: Double
- Definition Classes
- LightGBMFractionParams
-
def
getFeatureFractionSeed: Int
- Definition Classes
- LightGBMSeedParams
-
final
def
getFeaturesCol: String
- Definition Classes
- HasFeaturesCol
-
def
getFeaturesShapCol: String
- Definition Classes
- LightGBMPredictionParams
-
def
getGroupCol: String
- Definition Classes
- HasGroupCol
-
def
getImprovementTolerance: Double
- Definition Classes
- LightGBMLearnerParams
-
def
getInitScoreCol: String
- Definition Classes
- HasInitScoreCol
-
def
getIsEnableSparse: Boolean
- Definition Classes
- LightGBMDatasetParams
-
def
getIsProvideTrainingMetric: Boolean
- Definition Classes
- LightGBMParams
-
final
def
getLabelCol: String
- Definition Classes
- HasLabelCol
- def getLabelGain: Array[Double]
-
def
getLambdaL1: Double
- Definition Classes
- LightGBMParams
-
def
getLambdaL2: Double
- Definition Classes
- LightGBMParams
-
def
getLeafPredictionCol: String
- Definition Classes
- LightGBMPredictionParams
-
def
getLearningRate: Double
- Definition Classes
- LightGBMParams
-
def
getMatrixType: String
- Definition Classes
- LightGBMExecutionParams
-
def
getMaxBin: Int
- Definition Classes
- LightGBMBinParams
-
def
getMaxBinByFeature: Array[Int]
- Definition Classes
- LightGBMParams
-
def
getMaxCatThreshold: Int
- Definition Classes
- LightGBMCategoricalParams
-
def
getMaxCatToOnehot: Int
- Definition Classes
- LightGBMCategoricalParams
-
def
getMaxDeltaStep: Double
- Definition Classes
- LightGBMParams
-
def
getMaxDepth: Int
- Definition Classes
- LightGBMParams
-
def
getMaxDrop: Int
- Definition Classes
- LightGBMDartParams
-
def
getMaxNumClasses: Int
- Definition Classes
- LightGBMParams
- def getMaxPosition: Int
-
def
getMaxStreamingOMPThreads: Int
- Definition Classes
- LightGBMExecutionParams
-
def
getMetric: String
- Definition Classes
- LightGBMParams
-
def
getMicroBatchSize: Int
- Definition Classes
- LightGBMExecutionParams
-
def
getMinDataInLeaf: Int
- Definition Classes
- LightGBMParams
-
def
getMinDataPerBin: Int
- Definition Classes
- LightGBMParams
-
def
getMinDataPerGroup: Int
- Definition Classes
- LightGBMCategoricalParams
-
def
getMinGainToSplit: Double
- Definition Classes
- LightGBMParams
-
def
getMinSumHessianInLeaf: Double
- Definition Classes
- LightGBMParams
-
def
getModel(trainParams: BaseTrainParams, lightGBMBooster: LightGBMBooster): LightGBMRankerModel
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
- LightGBMRanker → LightGBMBase
-
def
getModelString: String
- Definition Classes
- LightGBMParams
-
def
getMonotoneConstraints: Array[Int]
- Definition Classes
- LightGBMLearnerParams
-
def
getMonotoneConstraintsMethod: String
- Definition Classes
- LightGBMLearnerParams
-
def
getMonotonePenalty: Double
- Definition Classes
- LightGBMLearnerParams
-
def
getNegBaggingFraction: Double
- Definition Classes
- LightGBMFractionParams
-
def
getNumBatches: Int
- Definition Classes
- LightGBMExecutionParams
-
def
getNumIterations: Int
- Definition Classes
- LightGBMParams
-
def
getNumLeaves: Int
- Definition Classes
- LightGBMParams
-
def
getNumTasks: Int
- Definition Classes
- LightGBMExecutionParams
-
def
getNumThreads: Int
- Definition Classes
- LightGBMExecutionParams
-
def
getObjective: String
- Definition Classes
- LightGBMObjectiveParams
-
def
getObjectiveSeed: Int
- Definition Classes
- LightGBMSeedParams
-
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
- Definition Classes
- LightGBMRanker → LightGBMBase
-
final
def
getOrDefault[T](param: Param[T]): T
- Definition Classes
- Params
-
def
getOtherRate: Double
- Definition Classes
- LightGBMLearnerParams
-
def
getParallelism: String
- Definition Classes
- LightGBMExecutionParams
-
def
getParam(paramName: String): Param[Any]
- Definition Classes
- Params
-
def
getParamInfo(p: Param[_]): ParamInfo[_]
- Definition Classes
- BaseWrappable
-
def
getPassThroughArgs: String
- Definition Classes
- LightGBMExecutionParams
-
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
-
def
getPosBaggingFraction: Double
- Definition Classes
- LightGBMFractionParams
-
def
getPredictDisableShapeCheck: Boolean
- Definition Classes
- LightGBMPredictionParams
-
final
def
getPredictionCol: String
- Definition Classes
- HasPredictionCol
-
def
getReferenceDataset: Array[Byte]
- Definition Classes
- LightGBMExecutionParams
-
def
getRepartitionByGroupingColumn: Boolean
- Definition Classes
- LightGBMExecutionParams
-
def
getSamplingMode: String
- Definition Classes
- LightGBMExecutionParams
-
def
getSamplingSubsetSize: Int
- Definition Classes
- LightGBMExecutionParams
-
def
getSeed: Int
- Definition Classes
- LightGBMSeedParams
-
def
getSkipDrop: Double
- Definition Classes
- LightGBMDartParams
-
def
getSlotNames: Array[String]
- Definition Classes
- LightGBMSlotParams
-
def
getTimeout: Double
- Definition Classes
- LightGBMExecutionParams
-
def
getTopK: Int
- Definition Classes
- LightGBMExecutionParams
-
def
getTopRate: Double
- Definition Classes
- LightGBMLearnerParams
-
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
- LightGBMRanker → LightGBMBase
-
def
getUniformDrop: Boolean
- Definition Classes
- LightGBMDartParams
-
def
getUseBarrierExecutionMode: Boolean
- Definition Classes
- LightGBMExecutionParams
-
def
getUseMissing: Boolean
- Definition Classes
- LightGBMDatasetParams
-
def
getUseSingleDatasetMode: Boolean
- Definition Classes
- LightGBMExecutionParams
-
def
getValidationIndicatorCol: String
- Definition Classes
- HasValidationIndicatorCol
-
def
getVerbosity: Int
- Definition Classes
- LightGBMParams
-
def
getWeightCol: String
- Definition Classes
- HasWeightCol
-
def
getXGBoostDartMode: Boolean
- Definition Classes
- LightGBMDartParams
-
def
getZeroAsMissing: Boolean
- Definition Classes
- LightGBMDatasetParams
-
val
groupCol: Param[String]
The name of the group column
The name of the group column
- Definition Classes
- HasGroupCol
-
final
def
hasDefault[T](param: Param[T]): Boolean
- Definition Classes
- Params
-
def
hasParam(paramName: String): Boolean
- Definition Classes
- Params
-
val
improvementTolerance: DoubleParam
- Definition Classes
- LightGBMLearnerParams
-
val
initScoreCol: Param[String]
The name of the initial score column
The name of the initial score column
- Definition Classes
- HasInitScoreCol
-
final
def
isDefined(param: Param[_]): Boolean
- Definition Classes
- Params
-
val
isEnableSparse: BooleanParam
- Definition Classes
- LightGBMDatasetParams
-
val
isProvideTrainingMetric: BooleanParam
- Definition Classes
- LightGBMParams
-
final
def
isSet(param: Param[_]): Boolean
- Definition Classes
- Params
-
final
val
labelCol: Param[String]
- Definition Classes
- HasLabelCol
- val labelGain: DoubleArrayParam
-
val
lambdaL1: DoubleParam
- Definition Classes
- LightGBMParams
-
val
lambdaL2: DoubleParam
- Definition Classes
- LightGBMParams
-
val
leafPredictionCol: Param[String]
- Definition Classes
- LightGBMPredictionParams
-
val
learningRate: DoubleParam
- Definition Classes
- LightGBMParams
-
def
logClass(featureName: String): Unit
- Definition Classes
- SynapseMLLogging
-
def
logFit[T](f: ⇒ T, columns: Int): T
- Definition Classes
- SynapseMLLogging
-
def
logTransform[T](f: ⇒ T, columns: Int): T
- Definition Classes
- SynapseMLLogging
-
def
logVerb[T](verb: String, f: ⇒ T, columns: Option[Int] = None): T
- Definition Classes
- SynapseMLLogging
-
def
makeDotnetFile(conf: CodegenConfig): Unit
- Definition Classes
- DotnetWrappable
-
def
makePyFile(conf: CodegenConfig): Unit
- Definition Classes
- PythonWrappable
-
def
makeRFile(conf: CodegenConfig): Unit
- Definition Classes
- RWrappable
-
val
matrixType: Param[String]
- Definition Classes
- LightGBMExecutionParams
-
val
maxBin: IntParam
- Definition Classes
- LightGBMBinParams
-
val
maxBinByFeature: IntArrayParam
- Definition Classes
- LightGBMParams
-
val
maxCatThreshold: IntParam
- Definition Classes
- LightGBMCategoricalParams
-
val
maxCatToOnehot: IntParam
- Definition Classes
- LightGBMCategoricalParams
-
val
maxDeltaStep: DoubleParam
- Definition Classes
- LightGBMParams
-
val
maxDepth: IntParam
- Definition Classes
- LightGBMParams
-
val
maxDrop: IntParam
- Definition Classes
- LightGBMDartParams
-
val
maxNumClasses: IntParam
- Definition Classes
- LightGBMParams
- val maxPosition: IntParam
-
val
maxStreamingOMPThreads: IntParam
- Definition Classes
- LightGBMExecutionParams
-
val
metric: Param[String]
- Definition Classes
- LightGBMParams
-
val
microBatchSize: IntParam
- Definition Classes
- LightGBMExecutionParams
-
val
minDataInLeaf: IntParam
- Definition Classes
- LightGBMParams
-
val
minDataPerBin: IntParam
- Definition Classes
- LightGBMParams
-
val
minDataPerGroup: IntParam
- Definition Classes
- LightGBMCategoricalParams
-
val
minGainToSplit: DoubleParam
- Definition Classes
- LightGBMParams
-
val
minSumHessianInLeaf: DoubleParam
- Definition Classes
- LightGBMParams
-
val
modelString: Param[String]
- Definition Classes
- LightGBMParams
-
val
monotoneConstraints: IntArrayParam
- Definition Classes
- LightGBMLearnerParams
-
val
monotoneConstraintsMethod: Param[String]
- Definition Classes
- LightGBMLearnerParams
-
val
monotonePenalty: DoubleParam
- Definition Classes
- LightGBMLearnerParams
-
val
negBaggingFraction: DoubleParam
- Definition Classes
- LightGBMFractionParams
-
val
numBatches: IntParam
- Definition Classes
- LightGBMExecutionParams
-
val
numIterations: IntParam
- Definition Classes
- LightGBMParams
-
val
numLeaves: IntParam
- Definition Classes
- LightGBMParams
-
val
numTasks: IntParam
- Definition Classes
- LightGBMExecutionParams
-
val
numThreads: IntParam
- Definition Classes
- LightGBMExecutionParams
-
val
objective: Param[String]
- Definition Classes
- LightGBMObjectiveParams
-
val
objectiveSeed: IntParam
- Definition Classes
- LightGBMSeedParams
-
val
otherRate: DoubleParam
- Definition Classes
- LightGBMLearnerParams
-
val
parallelism: Param[String]
- Definition Classes
- LightGBMExecutionParams
-
lazy val
params: Array[Param[_]]
- Definition Classes
- Params
-
val
passThroughArgs: Param[String]
- Definition Classes
- LightGBMExecutionParams
-
val
posBaggingFraction: DoubleParam
- Definition Classes
- LightGBMFractionParams
-
val
predictDisableShapeCheck: BooleanParam
- Definition Classes
- LightGBMPredictionParams
-
final
val
predictionCol: Param[String]
- Definition Classes
- HasPredictionCol
-
def
prepareDataframe(dataset: Dataset[_], numTasks: Int): DataFrame
- Definition Classes
- LightGBMRanker → LightGBMBase
-
def
preprocessData(df: DataFrame): DataFrame
For Ranking, we need to sort the data within partitions by group prior to training to ensure training succeeds.
For Ranking, we need to sort the data within partitions by group prior to training to ensure training succeeds.
- df
The data frame to preprocess prior to training.
- returns
The preprocessed data, sorted within partition by group.
- Definition Classes
- LightGBMRanker → LightGBMBase
-
def
pyAdditionalMethods: String
- Definition Classes
- PythonWrappable
-
def
pyInitFunc(): String
- Definition Classes
- PythonWrappable
-
val
referenceDataset: ByteArrayParam
- Definition Classes
- LightGBMExecutionParams
-
val
repartitionByGroupingColumn: BooleanParam
- Definition Classes
- LightGBMExecutionParams
-
val
samplingMode: Param[String]
- Definition Classes
- LightGBMExecutionParams
-
val
samplingSubsetSize: IntParam
- Definition Classes
- LightGBMExecutionParams
-
def
save(path: String): Unit
- Definition Classes
- MLWritable
- Annotations
- @Since( "1.6.0" ) @throws( ... )
-
val
seed: IntParam
- Definition Classes
- LightGBMSeedParams
-
final
def
set[T](param: Param[T], value: T): LightGBMRanker.this.type
- Definition Classes
- Params
-
def
setBaggingFraction(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMFractionParams
-
def
setBaggingFreq(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setBaggingSeed(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMSeedParams
-
def
setBinSampleCount(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMBinParams
-
def
setBoostFromAverage(value: Boolean): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setBoostingType(value: String): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setCatSmooth(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMCategoricalParams
-
def
setCategoricalSlotIndexes(value: Array[Int]): LightGBMRanker.this.type
- Definition Classes
- LightGBMSlotParams
-
def
setCategoricalSlotNames(value: Array[String]): LightGBMRanker.this.type
- Definition Classes
- LightGBMSlotParams
-
def
setCatl2(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMCategoricalParams
-
def
setChunkSize(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setDataRandomSeed(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMSeedParams
-
def
setDataTransferMode(value: String): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setDefaultListenPort(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setDelegate(delegate: LightGBMDelegate): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setDeterministic(value: Boolean): LightGBMRanker.this.type
- Definition Classes
- LightGBMSeedParams
-
def
setDriverListenPort(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setDropRate(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMDartParams
-
def
setDropSeed(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMSeedParams
-
def
setEarlyStoppingRound(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMLearnerParams
- def setEvalAt(value: Array[Int]): LightGBMRanker.this.type
-
def
setExtraSeed(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMSeedParams
-
def
setFObj(value: FObjTrait): LightGBMRanker.this.type
- Definition Classes
- LightGBMObjectiveParams
-
def
setFeatureFraction(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMFractionParams
-
def
setFeatureFractionByNode(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMFractionParams
-
def
setFeatureFractionSeed(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMSeedParams
-
def
setFeaturesCol(value: String): LightGBMRanker
- Definition Classes
- Predictor
-
def
setFeaturesShapCol(value: String): LightGBMRanker.this.type
- Definition Classes
- LightGBMPredictionParams
-
def
setGroupCol(value: String): LightGBMRanker.this.type
- Definition Classes
- HasGroupCol
-
def
setImprovementTolerance(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMLearnerParams
-
def
setInitScoreCol(value: String): LightGBMRanker.this.type
- Definition Classes
- HasInitScoreCol
-
def
setIsEnableSparse(value: Boolean): LightGBMRanker.this.type
- Definition Classes
- LightGBMDatasetParams
-
def
setIsProvideTrainingMetric(value: Boolean): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setLabelCol(value: String): LightGBMRanker
- Definition Classes
- Predictor
- def setLabelGain(value: Array[Double]): LightGBMRanker.this.type
-
def
setLambdaL1(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setLambdaL2(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setLeafPredictionCol(value: String): LightGBMRanker.this.type
- Definition Classes
- LightGBMPredictionParams
-
def
setLearningRate(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setMatrixType(value: String): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setMaxBin(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMBinParams
-
def
setMaxBinByFeature(value: Array[Int]): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setMaxCatThreshold(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMCategoricalParams
-
def
setMaxCatToOnehot(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMCategoricalParams
-
def
setMaxDeltaStep(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setMaxDepth(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setMaxDrop(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMDartParams
-
def
setMaxNumClasses(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
- def setMaxPosition(value: Int): LightGBMRanker.this.type
-
def
setMaxStreamingOMPThreads(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setMetric(value: String): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setMicroBatchSize(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setMinDataInLeaf(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setMinDataPerBin(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setMinDataPerGroup(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMCategoricalParams
-
def
setMinGainToSplit(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setMinSumHessianInLeaf(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setModelString(value: String): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setMonotoneConstraints(value: Array[Int]): LightGBMRanker.this.type
- Definition Classes
- LightGBMLearnerParams
-
def
setMonotoneConstraintsMethod(value: String): LightGBMRanker.this.type
- Definition Classes
- LightGBMLearnerParams
-
def
setMonotonePenalty(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMLearnerParams
-
def
setNegBaggingFraction(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMFractionParams
-
def
setNumBatches(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setNumIterations(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setNumLeaves(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setNumTasks(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setNumThreads(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setObjective(value: String): LightGBMRanker.this.type
- Definition Classes
- LightGBMObjectiveParams
-
def
setObjectiveSeed(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMSeedParams
-
def
setOtherRate(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMLearnerParams
-
def
setParallelism(value: String): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setPassThroughArgs(value: String): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setPosBaggingFraction(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMFractionParams
-
def
setPredictDisableShapeCheck(value: Boolean): LightGBMRanker.this.type
- Definition Classes
- LightGBMPredictionParams
-
def
setPredictionCol(value: String): LightGBMRanker
- Definition Classes
- Predictor
-
def
setReferenceDataset(value: Array[Byte]): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setRepartitionByGroupingColumn(value: Boolean): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setSamplingMode(value: String): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setSamplingSubsetSize(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setSeed(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMSeedParams
-
def
setSkipDrop(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMDartParams
-
def
setSlotNames(value: Array[String]): LightGBMRanker.this.type
- Definition Classes
- LightGBMSlotParams
-
def
setTimeout(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setTopK(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setTopRate(value: Double): LightGBMRanker.this.type
- Definition Classes
- LightGBMLearnerParams
-
def
setUniformDrop(value: Boolean): LightGBMRanker.this.type
- Definition Classes
- LightGBMDartParams
-
def
setUseBarrierExecutionMode(value: Boolean): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setUseMissing(value: Boolean): LightGBMRanker.this.type
- Definition Classes
- LightGBMDatasetParams
-
def
setUseSingleDatasetMode(value: Boolean): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
-
def
setValidationIndicatorCol(value: String): LightGBMRanker.this.type
- Definition Classes
- HasValidationIndicatorCol
-
def
setVerbosity(value: Int): LightGBMRanker.this.type
- Definition Classes
- LightGBMParams
-
def
setWeightCol(value: String): LightGBMRanker.this.type
- Definition Classes
- HasWeightCol
-
def
setXGBoostDartMode(value: Boolean): LightGBMRanker.this.type
- Definition Classes
- LightGBMDartParams
-
def
setZeroAsMissing(value: Boolean): LightGBMRanker.this.type
- Definition Classes
- LightGBMDatasetParams
-
val
skipDrop: DoubleParam
- Definition Classes
- LightGBMDartParams
-
val
slotNames: StringArrayParam
- Definition Classes
- LightGBMSlotParams
-
def
stringFromTrainedModel(model: LightGBMRankerModel): String
- Definition Classes
- LightGBMRanker → LightGBMBase
-
val
timeout: DoubleParam
- Definition Classes
- LightGBMExecutionParams
-
def
toString(): String
- Definition Classes
- Identifiable → AnyRef → Any
-
val
topK: IntParam
- Definition Classes
- LightGBMExecutionParams
-
val
topRate: DoubleParam
- Definition Classes
- LightGBMLearnerParams
-
def
transformSchema(schema: StructType): StructType
- Definition Classes
- Predictor → PipelineStage
-
val
uid: String
- Definition Classes
- LightGBMRanker → SynapseMLLogging → Identifiable
-
val
uniformDrop: BooleanParam
- Definition Classes
- LightGBMDartParams
-
val
useBarrierExecutionMode: BooleanParam
- Definition Classes
- LightGBMExecutionParams
-
val
useMissing: BooleanParam
- Definition Classes
- LightGBMDatasetParams
-
val
useSingleDatasetMode: BooleanParam
- Definition Classes
- LightGBMExecutionParams
-
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
-
val
weightCol: Param[String]
The name of the weight column
The name of the weight column
- Definition Classes
- HasWeightCol
-
def
write: MLWriter
- Definition Classes
- ComplexParamsWritable → MLWritable
-
val
xGBoostDartMode: BooleanParam
- Definition Classes
- LightGBMDartParams
-
val
zeroAsMissing: BooleanParam
- Definition Classes
- LightGBMDatasetParams
Deprecated Value Members
-
val
executionMode: Param[String]
- Definition Classes
- LightGBMExecutionParams
- Annotations
- @deprecated
- Deprecated
(Since version 0.11.1) Please use 'dataTransferMode'
-
def
setExecutionMode(value: String): LightGBMRanker.this.type
- Definition Classes
- LightGBMExecutionParams
- Annotations
- @deprecated
- Deprecated
(Since version 0.11.1) Please use 'setDataTransferMode'