Advertisement
Guest User

Untitled

a guest
Aug 24th, 2016
62
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Scala 0.77 KB | None | 0 0
  1.   private def precisionAtK(k: Int, actual: Array[Int], predicted: Array[Int]): Double = {
  2.     //val pred = predicted.take(k)
  3.     val a = actual.length
  4.     //actual.intersect(pred).length / a : Double
  5.     a
  6.   }
  7.  
  8.   def averagePrecisionAtK(k: Int) = {
  9.     new Score[Array[Int]] with PerformanceScore {
  10.       //TODO
  11.       override def evaluate(data: DataSet[(Array[Int], Array[Int])]): DataSet[Double] = {
  12.         val perUser = data.map { trueAndPred => precisionAtK(k, trueAndPred._1, trueAndPred._2)
  13.           //val pred = trueAndPred._2.take(k)
  14.           //val a = trueAndPred._1.length
  15.           //trueAndPred._1.intersect(pred).length / a : Double}
  16.           //a
  17.         }
  18.           //val perUser = data.map(trueAndPred => 2)
  19.         perUser.mean()
  20.       }
  21.     }
  22.   }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement