ierhon commited on
Commit
b3a8af2
1 Parent(s): 16d5400

Update word_emb.py

Browse files
Files changed (1) hide show
  1. word_emb.py +1 -1
word_emb.py CHANGED
@@ -10,6 +10,6 @@ tokenizer.fit_on_texts(lines)
10
  sequences = tokenizer.texts_to_sequences(lines)
11
  tokens = [[str(i) for i in seq] for seq in sequences]
12
 
13
- model = Word2Vec(tokens, window=3, min_count=1, vector_size=emb_o_dim)
14
 
15
  model.save("w2v.model")
 
10
  sequences = tokenizer.texts_to_sequences(lines)
11
  tokens = [[str(i) for i in seq] for seq in sequences]
12
 
13
+ model = Word2Vec(tokens, window=6, min_count=1, vector_size=emb_o_dim)
14
 
15
  model.save("w2v.model")