Update WordVectorTraining.java

vijay033 · Feb 16, 2019 · aaec1f8 · aaec1f8
1 parent a6462d1
commit aaec1f8
Showing 1 changed file with 6 additions and 5 deletions.
diff --git 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)