LogRegRanker

io.github.metarank.ltrlib.ranking.pointwise.LogRegRanker
See theLogRegRanker companion object
case class LogRegRanker(train: Dataset, options: RegressionOptions) extends Ranker[LogRegModel]

Attributes

Companion:
object
Graph
Supertypes
trait Serializable
trait Product
trait Equals
trait Logging
class Object
trait Matchable
class Any

Members list

Concise view

Value members

Concrete methods

override def fit(): LogRegModel

Attributes

Definition Classes
def predict(row: RealVector, weights: RealVector, intercept: Double): Double
def prepare(): (Array2DRowRealMatrix, ArrayRealVector)
def randomSample(from: Int, to: Int, count: Int): Array[Int]
def trainBatchSGD(x: Array2DRowRealMatrix, y: RealVector, iterations: Int, batchSize: Int, lr: Double): RegWeights
def trainSGD(x: Array2DRowRealMatrix, y: RealVector, iterations: Int, lr: Double): RegWeights

Inherited methods

def productElementNames: Iterator[String]

Attributes

Inherited from:
Product
def productIterator: Iterator[Any]

Attributes

Inherited from:
Product

Concrete fields

val x: Array2DRowRealMatrix
val y: ArrayRealVector

Inherited fields

val logger: Logger

Attributes

Inherited from:
Logging