From cea15ab055c1156c0f1ab0f0ac6f4c6a435908a2 Mon Sep 17 00:00:00 2001
From: Erick Lavoie <erick.lavoie@epfl.ch>
Date: Mon, 11 Apr 2022 17:19:47 +0200
Subject: [PATCH] Added new required prediction

---
 src/main/scala/distributed/Exact.scala          | 3 ++-
 src/main/scala/optimizing/Optimizing.scala      | 3 ++-
 src/test/scala/distributed/ExactTests.scala     | 3 +++
 src/test/scala/optimizing/OptimizingTests.scala | 3 +++
 4 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/main/scala/distributed/Exact.scala b/src/main/scala/distributed/Exact.scala
index b2b6ebf..ae05d2d 100644
--- a/src/main/scala/distributed/Exact.scala
+++ b/src/main/scala/distributed/Exact.scala
@@ -90,7 +90,8 @@ object Exact {
             "2.knn_u1v864" -> ujson.Num(0.0),
             "3.knn_u1v886" -> ujson.Num(0.0),
             "4.PredUser1Item1" -> ujson.Num(0.0),
-            "5.Mae" -> ujson.Num(measurements(0)._1)
+            "5.PredUser327Item2" -> ujson.Num(0.0),
+            "6.Mae" -> ujson.Num(0.0)
           ),
           "EK.2" ->  ujson.Obj(
             "average (ms)" -> ujson.Num(mean(timings)), // Datatype of answer: Double
diff --git a/src/main/scala/optimizing/Optimizing.scala b/src/main/scala/optimizing/Optimizing.scala
index 0afcc5f..08d27c3 100644
--- a/src/main/scala/optimizing/Optimizing.scala
+++ b/src/main/scala/optimizing/Optimizing.scala
@@ -76,7 +76,8 @@ object Optimizing extends App {
             "2.k10u1v864" -> ujson.Num(0.0),
             "3.k10u1v886" -> ujson.Num(0.0),
             "4.PredUser1Item1" -> ujson.Num(0.0),
-            "5.Mae" -> ujson.Num(0.0)
+            "5.PredUser327Item2" -> ujson.Num(0.0),
+            "6.Mae" -> ujson.Num(0.0)
           ),
           "BR.2" ->  ujson.Obj(
             "average (ms)" -> ujson.Num(mean(timings)), // Datatype of answer: Double
diff --git a/src/test/scala/distributed/ExactTests.scala b/src/test/scala/distributed/ExactTests.scala
index ad94055..4a05e2a 100644
--- a/src/test/scala/distributed/ExactTests.scala
+++ b/src/test/scala/distributed/ExactTests.scala
@@ -46,6 +46,9 @@ class ExactTests extends AnyFunSuite with BeforeAndAfterAll {
      // Prediction user 1 and item 1
      assert(within(1.0, 0.0, 0.0001))
 
+     // Prediction user 327 and item 2
+     assert(within(1.0, 0.0, 0.0001))
+
      // MAE on test
      assert(within(1.0, 0.0, 0.0001)) 
    } 
diff --git a/src/test/scala/optimizing/OptimizingTests.scala b/src/test/scala/optimizing/OptimizingTests.scala
index b13b1c9..3ef3a72 100644
--- a/src/test/scala/optimizing/OptimizingTests.scala
+++ b/src/test/scala/optimizing/OptimizingTests.scala
@@ -41,6 +41,9 @@ class OptimizingTests extends AnyFunSuite with BeforeAndAfterAll {
      // Prediction user 1 and item 1
      assert(within(1.0, 0.0, 0.0001))
 
+     // Prediction user 327 and item 2
+     assert(within(1.0, 0.0, 0.0001))
+
      // MAE on test2
      assert(within(1.0, 0.0, 0.0001)) 
    } 
-- 
GitLab