All Downloads are FREE. Search and download functionalities are using the official Maven repository.

io.github.metarank.ltrlib.booster.XGBoostOptions.scala Maven / Gradle / Ivy

There is a newer version: 0.2.6
Show newest version
package io.github.metarank.ltrlib.booster

import io.github.metarank.ltrlib.booster.Booster.BoosterOptions

import scala.util.Random

case class XGBoostOptions(
    trees: Int = 100,
    learningRate: Double = 0.1,
    ndcgCutoff: Int = 10,
    maxDepth: Int = 8,
    randomSeed: Int = Random.nextInt(),
    subsample: Double = 1.0
) extends BoosterOptions




© 2015 - 2024 Weber Informatics LLC | Privacy Policy