trait LightGBMModelMethods extends LightGBMModelParams with Logging
Contains common LightGBM model methods across all LightGBM learner types.
- Alphabetic
- By Inheritance
- LightGBMModelMethods
- Logging
- LightGBMModelParams
- Wrappable
- RWrappable
- PythonWrappable
- BaseWrappable
- Params
- Serializable
- Serializable
- Identifiable
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Abstract Value Members
Concrete Value Members
-
final
def
clear(param: Param[_]): LightGBMModelMethods.this.type
- Definition Classes
- Params
-
def
explainParam(param: Param[_]): String
- Definition Classes
- Params
-
def
explainParams(): String
- Definition Classes
- Params
-
final
def
extractParamMap(): ParamMap
- Definition Classes
- Params
-
final
def
extractParamMap(extra: ParamMap): ParamMap
- Definition Classes
- Params
-
final
def
get[T](param: Param[T]): Option[T]
- Definition Classes
- Params
-
def
getBoosterBestIteration(): Int
Public method to get the best iteration from the booster.
Public method to get the best iteration from the booster.
- returns
The best iteration, if early stopping was triggered.
-
def
getBoosterNumClasses(): Int
Public method to get the number of classes from the booster.
Public method to get the number of classes from the booster.
- returns
The number of classes.
-
def
getBoosterNumFeatures(): Int
Public method to get the number of features from the booster.
Public method to get the number of features from the booster.
- returns
The number of features.
-
def
getBoosterNumTotalIterations(): Int
Public method to get the total number of iterations trained.
Public method to get the total number of iterations trained.
- returns
The total number of iterations trained.
-
def
getBoosterNumTotalModel(): Int
Public method to get the total number of models trained.
Public method to get the total number of models trained. Note this may be larger than the number of iterations, since in multiclass a model is trained per class for each iteration.
- returns
The total number of models.
-
final
def
getDefault[T](param: Param[T]): Option[T]
- Definition Classes
- Params
-
def
getDenseFeatureShaps(features: Array[Double]): Array[Double]
Public method for pyspark API to get the dense local SHAP feature importance values for an instance.
Public method for pyspark API to get the dense local SHAP feature importance values for an instance.
- features
The local instance or row to compute the SHAP values for.
- returns
The local feature importance values.
-
def
getFeatureImportances(importanceType: String): Array[Double]
Public method to get the global feature importance values.
Public method to get the global feature importance values.
- importanceType
split or gini
- returns
The global feature importance values.
-
def
getFeatureShaps(features: Vector): Array[Double]
Public method to get the vector local SHAP feature importance values for an instance.
Public method to get the vector local SHAP feature importance values for an instance.
- features
The local instance or row to compute the SHAP values for.
- returns
The local feature importance values.
-
def
getLightGBMBooster: LightGBMBooster
- Definition Classes
- LightGBMModelParams
-
def
getModel: LightGBMBooster
Alias for same method
-
def
getNativeModel(): String
Gets the native model serialized representation as a string.
-
def
getNumIterations: Int
- Definition Classes
- LightGBMModelParams
-
final
def
getOrDefault[T](param: Param[T]): T
- Definition Classes
- Params
-
def
getParam(paramName: String): Param[Any]
- Definition Classes
- Params
-
def
getParamInfo(p: Param[_]): ParamInfo[_]
- Definition Classes
- BaseWrappable
-
def
getSparseFeatureShaps(size: Int, indices: Array[Int], values: Array[Double]): Array[Double]
Public method for pyspark API to get the sparse local SHAP feature importance values for an instance.
Public method for pyspark API to get the sparse local SHAP feature importance values for an instance.
- returns
The local feature importance values.
-
def
getStartIteration: Int
- Definition Classes
- LightGBMModelParams
-
final
def
hasDefault[T](param: Param[T]): Boolean
- Definition Classes
- Params
-
def
hasParam(paramName: String): Boolean
- Definition Classes
- Params
-
final
def
isDefined(param: Param[_]): Boolean
- Definition Classes
- Params
-
final
def
isSet(param: Param[_]): Boolean
- Definition Classes
- Params
-
val
lightGBMBooster: LightGBMBoosterParam
- Definition Classes
- LightGBMModelParams
-
def
makePyFile(conf: CodegenConfig): Unit
- Definition Classes
- PythonWrappable
-
def
makeRFile(conf: CodegenConfig): Unit
- Definition Classes
- RWrappable
-
val
numIterations: IntParam
- Definition Classes
- LightGBMModelParams
-
lazy val
params: Array[Param[_]]
- Definition Classes
- Params
-
def
pyAdditionalMethods: String
- Definition Classes
- PythonWrappable
-
def
pyInitFunc(): String
- Definition Classes
- PythonWrappable
-
def
saveNativeModel(filename: String, overwrite: Boolean): Unit
Saves the native model serialized representation to file.
Saves the native model serialized representation to file.
- filename
The name of the file to save the model to
- overwrite
Whether to overwrite if the file already exists
-
final
def
set[T](param: Param[T], value: T): LightGBMModelMethods.this.type
- Definition Classes
- Params
-
def
setLightGBMBooster(value: LightGBMBooster): LightGBMModelMethods.this.type
- Definition Classes
- LightGBMModelParams
-
def
setNumIterations(value: Int): LightGBMModelMethods.this.type
- Definition Classes
- LightGBMModelParams
-
def
setStartIteration(value: Int): LightGBMModelMethods.this.type
- Definition Classes
- LightGBMModelParams
-
val
startIteration: IntParam
- Definition Classes
- LightGBMModelParams
-
def
toString(): String
- Definition Classes
- Identifiable → AnyRef → Any