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 LightGBMBooster(model: LGBMBooster, datasets: Map[LGBMDataset, Int] = mutable.Map.empty) extends Booster[LGBMDataset] with Product with Serializable
  4. case class LightGBMOptions(trees: Int = 100, learningRate: Double = 0.1, ndcgCutoff: Int = 10, maxDepth: Int = 8, randomSeed: Int = Random.nextInt(), numLeaves: Int = 16) extends BoosterOptions with Product with Serializable
  5. case class XGBoostBooster(model: ml.dmlc.xgboost4j.java.Booster) extends Booster[DMatrix] with Product with Serializable
  6. case class XGBoostOptions(trees: Int = 100, learningRate: Double = 0.1, ndcgCutoff: Int = 10, maxDepth: Int = 8, randomSeed: Int = Random.nextInt()) extends BoosterOptions with Product with Serializable

Value Members

  1. object Booster
  2. object LightGBMBooster extends BoosterFactory[LGBMDataset, LightGBMBooster, LightGBMOptions] with Serializable
  3. object XGBoostBooster extends BoosterFactory[DMatrix, XGBoostBooster, XGBoostOptions] with Serializable

Ungrouped