• Home
• io.github.metarank
• ltrlib_3
• 0.1.19
• source code
• MRR.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

io.github.metarank.ltrlib.metric.MRR.scala Maven / Gradle / Ivy

package io.github.metarank.ltrlib.metric

object MRR extends Metric {
  override def eval(y: Array[Array[Double]], yhat: Array[Array[Double]]): Double = {
    var mrr = 0.0
    for {
      group <- y.indices
    } {
      if (y(group).exists(_ != 0.0)) {
        val zipped     = y(group).zip(yhat(group))
        val firstClick = zipped.sortBy(-_._2).map(_._1).indexWhere(_ > 0.0)
        mrr += 1.0 / (firstClick + 1)
      }
    }
    mrr / y.length
  }
}