io.github.metarank.ltrlib.booster
Members list
Concise view
Type members
Classlikes
Attributes
- Companion:
- object
- Graph
- Supertypes
- Known subtypes
- class LightGBMBoosterclass XGBoostBooster
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
case class LightGBMBooster(model: LGBMBooster, datasets: Map[LGBMDataset, Int]) extends Booster[LGBMDataset]
Attributes
- Companion:
- class
- Graph
- Supertypes
- trait Producttrait Mirrorclass Objecttrait Matchableclass Any
- Self type
- LightGBMBooster.type
case class LightGBMOptions(trees: Int, learningRate: Double, ndcgCutoff: Int, maxDepth: Int, randomSeed: Int, numLeaves: Int, featureFraction: Double) extends BoosterOptions
Attributes
- Graph
- Supertypes
Attributes
- Companion:
- class
- Graph
- Supertypes
- trait Producttrait Mirrorclass Objecttrait Matchableclass Any
- Self type
- XGBoostBooster.type
case class XGBoostOptions(trees: Int, learningRate: Double, ndcgCutoff: Int, maxDepth: Int, randomSeed: Int, subsample: Double) extends BoosterOptions
Attributes
- Graph
- Supertypes