fix(word2vec): simple example of word2vec

2016-09-16 14:03:15 +04:30
parent d4ac90bbd5
commit 0d43814448
4 changed files with 12 additions and 7 deletions
--- a/examples/word2vec.hs
+++ b/examples/word2vec.hs
@@ -27,13 +27,14 @@ module Main where
              "the dwarf poisons the king", "the dwarf poisons the queen"]

    let session = def { learningRate = 0.1
-                      , batchSize = 16
+                      , batchSize = 1
                      , epochs = 100
+                      , debug = True
                      } :: Session
        w2v = def { docs = ds
                  , dimensions = 50
                  , method = SkipGram
-                  , window = 3
+                  , window = 2
                  } :: Word2Vec