commit 392f34769e3e523d8818aa5ac46e3a40de4b2390
author Naoki Takezoe <takezoe@apache.org> Wed Jul 19 10:13:10 2017 +0900
committer Naoki Takezoe <takezoe@apache.org> Wed Jul 19 10:13:10 2017 +0900
tree 1c6c52a31a2443411f80d9e1cdf02d82863ed941
parent a0585011e5a94a2198690c0f18c478b87c01fdd8

Fix use of case class

Closes #16

src/main/scala/ALSAlgorithm.scala

diff --git a/src/main/scala/ALSAlgorithm.scala b/src/main/scala/ALSAlgorithm.scala
index 7cdf6e9..435a9b5 100644
--- a/src/main/scala/ALSAlgorithm.scala
+++ b/src/main/scala/ALSAlgorithm.scala
@@ -168,13 +168,13 @@
     val topScores = getTopN(filteredScore, query.num)(ord).toArray
 
     val itemScores = topScores.map { case (i, s) =>
-      new ItemScore(
+      ItemScore(
         item = model.itemIntStringMap(i),
         score = s
       )
     }
 
-    new PredictedResult(itemScores)
+    PredictedResult(itemScores)
   }
 
   private

src/main/scala/CooccurrenceAlgorithm.scala

diff --git a/src/main/scala/CooccurrenceAlgorithm.scala b/src/main/scala/CooccurrenceAlgorithm.scala
index b56ecec..c6e0e08 100644
--- a/src/main/scala/CooccurrenceAlgorithm.scala
+++ b/src/main/scala/CooccurrenceAlgorithm.scala
@@ -129,7 +129,7 @@
         )
       }
 
-    new PredictedResult(itemScores)
+    PredictedResult(itemScores)
 
   }
 

src/main/scala/Engine.scala

diff --git a/src/main/scala/Engine.scala b/src/main/scala/Engine.scala
index c4a7459..df1e007 100644
--- a/src/main/scala/Engine.scala
+++ b/src/main/scala/Engine.scala
@@ -10,18 +10,18 @@
   categoryBlackList: Option[Set[String]],
   whiteList: Option[Set[String]],
   blackList: Option[Set[String]]
-) extends Serializable
+)
 
 case class PredictedResult(
   itemScores: Array[ItemScore]
-) extends Serializable {
+){
   override def toString: String = itemScores.mkString(",")
 }
 
 case class ItemScore(
   item: String,
   score: Double
-) extends Serializable
+)
 
 object SimilarProductEngine extends EngineFactory {
   def apply() = {