io.github.metarank.ltrlib.booster
Members list
Concise view
Type members
Classlikes
Attributes
- Companion:
- object
- Graph
- Supertypes
- Known subtypes
object Booster
case class BoosterDataset(original: Dataset, data: Array[Double], labels: Array[Double], groups: Array[Int], rows: Int, cols: Int, featureNames: Array[String])
Attributes
- Graph
- Supertypes
- trait Serializabletrait Producttrait Equalsclass Objecttrait Matchableclass Any
Attributes
- Companion:
- class
- Graph
- Supertypes
- trait Producttrait Mirrorclass Objecttrait Matchableclass Any
- Self type
- CatboostBooster.type
case class CatboostOptions(trees: Int, learningRate: Double, ndcgCutoff: Int, maxDepth: Int, randomSeed: Int, objective: String, loggingLevel: String, earlyStopping: Option[Int], boostingType: String) extends BoosterOptions
Attributes
- Graph
- Supertypes
object LightGBMBooster extends BoosterFactory[LGBMDataset, LightGBMBooster, LightGBMOptions] with Logging
Attributes
- Companion:
- class
- Graph
- Supertypes
- Self type
- LightGBMBooster.type
case class LightGBMOptions(trees: Int, learningRate: Double, ndcgCutoff: Int, maxDepth: Int, randomSeed: Int, numLeaves: Int, featureFraction: Double, earlyStopping: Option[Int]) extends BoosterOptions
Attributes
- Graph
- Supertypes
Attributes
- Companion:
- class
- Graph
- Supertypes
- Self type
- XGBoostBooster.type
case class XGBoostOptions(trees: Int, learningRate: Double, ndcgCutoff: Int, maxDepth: Int, randomSeed: Int, subsample: Double, earlyStopping: Option[Int], treeMethod: String) extends BoosterOptions
Attributes
- Graph
- Supertypes