From aaec1f8a919284253c8a887aa753df93115efd44 Mon Sep 17 00:00:00 2001
From: Vijayanand Kumar <44333153+vijay033@users.noreply.github.com>
Date: Sat, 16 Feb 2019 11:13:21 +0530
Subject: [PATCH] Update WordVectorTraining.java

---
 WordVectorTraining.java | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/WordVectorTraining.java b/WordVectorTraining.java
index f6adb2e..fabf038 100644
--- a/WordVectorTraining.java
+++ b/WordVectorTraining.java
@@ -139,10 +139,11 @@ public void run() {
     public void trainW2V(){
         SentenceIterator iterator = null;
         TokenizerFactory tokenizerFactory = null;
-        VocabCache<VocabWord> cache = null;
-        WeightLookupTable<VocabWord> table = null;
         word2Vec  = new Word2Vec[in_vectorStream.length];
 
+//         VocabCache<VocabWord> cache = null;
+//         WeightLookupTable<VocabWord> table = null;
+        
         if(wordVectorSaver.getSavedModelState() == false){
 
             new Thread(mMessageSender).start();
@@ -168,9 +169,9 @@ public void trainW2V(){
                         .seed(42)
                         .windowSize(5)
                         .epochs(1)
-//                        .batchSize(10)
-//                        .stopWords(stopwords)
-//                        .stopWords(extendedStopwords)
+                       .batchSize(10)
+                       .stopWords(stopwords)
+                       .stopWords(extendedStopwords)
                         .iterate(iterator)
                         .tokenizerFactory(tokenizerFactory)
 //                        .lookupTable(table)