ltrlib
API
Generated with
ltrlib/io.github.metarank.ltrlib/io.github.metarank.ltrlib.booster

io.github.metarank.ltrlib.booster

package io.github.metarank.ltrlib.booster

Members list

Type members

Classlikes

trait Booster[D] extends Model

Attributes

Companion
object
Supertypes
trait Model
class Object
trait Matchable
class Any
Known subtypes
object Booster

Attributes

Companion
trait
Supertypes
class Object
trait Matchable
class Any
Self type
Booster.type
case class BoosterDataset(original: Dataset, data: Array[Double], labels: Array[Double], groups: Array[Int], positions: Array[Int], rows: Int, cols: Int, featureNames: Array[String])

Attributes

Supertypes
trait Serializable
trait Product
trait Equals
class Object
trait Matchable
class Any
Show all
case class CatboostBooster(booster: CatBoostModel, bytes: Array[Byte]) extends Booster[String]

Attributes

Companion
object
Supertypes
trait Serializable
trait Product
trait Equals
trait Booster[String]
trait Model
class Object
trait Matchable
class Any
Show all
object CatboostBooster extends BoosterFactory[String, CatboostBooster, CatboostOptions]

Attributes

Companion
class
Supertypes
trait Product
trait Mirror
class Object
trait Matchable
class Any
Show all
Self 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

Supertypes
trait Serializable
trait Product
trait Equals
class Object
trait Matchable
class Any
Show all
case class LightGBMBooster(model: LGBMBooster) extends Booster[LGBMDataset], Logging

Attributes

Companion
object
Supertypes
trait Serializable
trait Product
trait Equals
trait Logging
trait Booster[LGBMDataset]
trait Model
class Object
trait Matchable
class Any
Show all
object LightGBMBooster extends BoosterFactory[LGBMDataset, LightGBMBooster, LightGBMOptions], Logging

Attributes

Companion
class
Supertypes
trait Product
trait Mirror
trait Logging
class Object
trait Matchable
class Any
Show all
Self type
case class LightGBMOptions(trees: Int, learningRate: Double, ndcgCutoff: Int, maxDepth: Int, randomSeed: Int, numLeaves: Int, featureFraction: Double, earlyStopping: Option[Int], debias: Boolean) extends BoosterOptions

Attributes

Supertypes
trait Serializable
trait Product
trait Equals
class Object
trait Matchable
class Any
Show all
case class XGBoostBooster(model: Booster, featureTypes: Array[String]) extends Booster[DMatrix], Logging

Attributes

Companion
object
Supertypes
trait Serializable
trait Product
trait Equals
trait Logging
trait Booster[DMatrix]
trait Model
class Object
trait Matchable
class Any
Show all
object XGBoostBooster extends BoosterFactory[DMatrix, XGBoostBooster, XGBoostOptions], Logging

Attributes

Companion
class
Supertypes
trait Product
trait Mirror
trait Logging
class Object
trait Matchable
class Any
Show all
Self type
case class XGBoostOptions(trees: Int, learningRate: Double, ndcgCutoff: Int, maxDepth: Int, randomSeed: Int, subsample: Double, earlyStopping: Option[Int], treeMethod: String, debias: Boolean) extends BoosterOptions

Attributes

Supertypes
trait Serializable
trait Product
trait Equals
class Object
trait Matchable
class Any
Show all
Generated with