CatboostBooster

io.github.metarank.ltrlib.booster.CatboostBooster
See theCatboostBooster companion object
case class CatboostBooster(booster: CatBoostModel, bytes: Array[Byte]) extends Booster[String]

Attributes

Companion:
object
Graph
Supertypes
trait Serializable
trait Product
trait Equals
trait Booster[String]
trait Model
class Object
trait Matchable
class Any

Members list

Concise view

Value members

Concrete methods

override def predictMat(values: Array[Double], rows: Int, cols: Int): Array[Double]

Attributes

Definition Classes
override def save(): Array[Byte]

Attributes

Definition Classes
override def weights(): Array[Double]

Attributes

Definition Classes

Inherited 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
Inherited from:
Booster
override def predict(values: RealVector): Double

Make single prediction

Make single prediction

Attributes

Definition Classes
Inherited from:
Booster
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
Inherited from:
Booster
def productElementNames: Iterator[String]

Attributes

Inherited from:
Product
def productIterator: Iterator[Any]

Attributes

Inherited from:
Product