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

package io.github.metarank.ltrlib.metric

import io.github.metarank.cfor._

object MSE extends Metric {
  def eval(y: Array[Array[Double]], yhat: Array[Array[Double]]): Double = {
    var sum = 0.0
    var cnt = 0
    cfor(y.indices) { i =>
      cfor(y(i).indices) { j =>
        {
          sum += (y(i)(j) - yhat(i)(j)) * (y(i)(j) - yhat(i)(j))
          cnt += 1
        }
      }

    }
    sum / cnt
  }
}