Retraining not required at inferencing time

gabrielmahia · Nov 29, 2018 · 14117be · 14117be
1 parent e33f9be
commit 14117be
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/Chapter05/TopicModelling/topicModellingwithLDA.scala b/Chapter05/TopicModelling/topicModellingwithLDA.scala
@@ -42,7 +42,7 @@ class LDAforTM() {
     .appName(s"OneVsRestExample")
     .getOrCreate()
 
-  def run(params: Params): Unit = {
+  def run(params: Params, flag:Boolean): Unit = {
     Logger.getRootLogger.setLevel(Level.WARN)
 
     // Load documents, and prepare them for LDA.
@@ -92,7 +92,8 @@ class LDAforTM() {
     val elapsed = (System.nanoTime() - startTime) / 1e9
 
     //Saving the model for future use
-    params.ldaModel.save(spark.sparkContext, "model/LDATrainedModel")
+    if(flag == true)
+      params.ldaModel.save(spark.sparkContext, "model/LDATrainedModel")   
 
     println("Finished training LDA model.  Summary:")
     println("Training time: " + elapsed + " sec")
@@ -196,4 +197,4 @@ object helperForLDA {
     }
     string.toString()
   }
-}
+}