Spaces:
Sleeping
Sleeping
Update word_emb.py
Browse files- 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=
|
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")
|