Merge pull request #10 from takezoe/mark-override
Mark override methods
diff --git a/src/main/scala/Algorithm.scala b/src/main/scala/Algorithm.scala
index 2e980f5..0a9d448 100644
--- a/src/main/scala/Algorithm.scala
+++ b/src/main/scala/Algorithm.scala
@@ -15,6 +15,7 @@
@transient lazy val logger = Logger[this.type]
+ override
def train(sc: SparkContext, data: PreparedData): Model = {
// Simply count number of events
// and multiple it by the algorithm parameter
@@ -23,6 +24,7 @@
new Model(mc = count)
}
+ override
def predict(model: Model, query: Query): PredictedResult = {
// Prefix the query with the model data
val result = s"${model.mc}-${query.q}"
diff --git a/src/main/scala/Preparator.scala b/src/main/scala/Preparator.scala
index 7e37500..18a8ed8 100644
--- a/src/main/scala/Preparator.scala
+++ b/src/main/scala/Preparator.scala
@@ -9,6 +9,7 @@
class Preparator
extends PPreparator[TrainingData, PreparedData] {
+ override
def prepare(sc: SparkContext, trainingData: TrainingData): PreparedData = {
new PreparedData(events = trainingData.events)
}