Booster

io.github.metarank.ltrlib.booster.Booster
See theBooster companion object
trait Booster[D] extends Model

Attributes

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

Members list

Value members

Abstract methods

def close(): Unit
def predictMat(values: Array[Double], rows: Int, cols: Int): Array[Double]
def save(): Array[Byte]
def weights(): Array[Double]

Concrete methods

override def eval(data: Dataset, metric: Metric): Double

Eval a metric over the whole dataset

Eval a metric over the whole dataset

Attributes

Definition Classes
def isClosed(): Boolean
override def predict(values: RealMatrix): ArrayRealVector

Make batch prediction, default impl falling back to per-row predict. You should overload it for better performance.

Make batch prediction, default impl falling back to per-row predict. You should overload it for better performance.

Attributes

Definition Classes
override def predict(values: RealVector): Double

Make single prediction

Make single prediction

Attributes

Definition Classes
def whenNotClosed[T](f: => T): T