Packages

package booster

Ordering
  1. Alphabetic
Visibility
  1. Public
  2. All

Type Members

  1. trait Booster[D] extends Model
  2. case class BoosterDataset(original: Dataset, data: Array[Double], labels: Array[Double], groups: Array[Int], rows: Int, cols: Int, featureNames: Array[String]) extends Product with Serializable
  3. case class CatboostBooster(booster: CatBoostModel, bytes: Array[Byte]) extends Booster[String] with Product with Serializable
  4. case class CatboostOptions(trees: Int = 100, learningRate: Double = 0.1, ndcgCutoff: Int = 10, maxDepth: Int = 8, randomSeed: Int = math.abs(Random.nextInt()), objective: String = "QueryRMSE", loggingLevel: String = "Verbose", earlyStopping: Option[Int] = None, boostingType: String = "Plain") extends BoosterOptions with Product with Serializable
  5. case class LightGBMBooster(model: LGBMBooster) extends Booster[LGBMDataset] with Logging with Product with Serializable
  6. case class LightGBMOptions(trees: Int = 100, learningRate: Double = 0.1, ndcgCutoff: Int = 10, maxDepth: Int = 8, randomSeed: Int = math.abs(Random.nextInt()), numLeaves: Int = 16, featureFraction: Double = 1.0, earlyStopping: Option[Int] = None) extends BoosterOptions with Product with Serializable
  7. case class XGBoostBooster(model: ml.dmlc.xgboost4j.java.Booster, featureTypes: Array[String]) extends Booster[DMatrix] with Logging with Product with Serializable
  8. case class XGBoostOptions(trees: Int = 100, learningRate: Double = 0.1, ndcgCutoff: Int = 10, maxDepth: Int = 8, randomSeed: Int = math.abs(Random.nextInt()), subsample: Double = 1.0, earlyStopping: Option[Int] = None, treeMethod: String = "exact") extends BoosterOptions with Product with Serializable

Value Members

  1. object Booster
  2. object CatboostBooster extends BoosterFactory[String, CatboostBooster, CatboostOptions] with Serializable
  3. object LightGBMBooster extends BoosterFactory[LGBMDataset, LightGBMBooster, LightGBMOptions] with Logging with Serializable
  4. object XGBoostBooster extends BoosterFactory[DMatrix, XGBoostBooster, XGBoostOptions] with Logging with Serializable

Ungrouped