• Home
• io.github.metarank
• ltrlib_3
• 0.1.19
• source code
• MAP.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.MAP.scala Maven / Gradle / Ivy

package io.github.metarank.ltrlib.metric

import io.github.metarank.cfor.cfor

case class MAP(cutoff: Int) extends Metric {
  override def eval(y: Array[Array[Double]], yhat: Array[Array[Double]]): Double = {
    var precision = 0.0
    cfor(y.indices) { group =>
      {
        val zipped   = y(group).zip(yhat(group)).sortBy(-_._2).take(cutoff)
        val relevant = zipped.count(_._1 > 0.0)
        precision += relevant.toDouble / cutoff
      }
    }
    precision / y.length
  }
}