Skip to content

Commit 89aebb4

Browse files
author
plliao
authored
Merge pull request #21 from Yoctol/add_vocab_size
add vocab size
2 parents f8bfd3e + 0d629af commit 89aebb4

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

seq2vec/word2vec/gensim_word2vec.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,3 +18,5 @@ def get_size(self):
1818
def get_index(self, key):
1919
return self.word2vec.vocab[key].index
2020

21+
def get_vocab_size(self):
22+
return len(self.word2vec.vocab)

0 commit comments

Comments
 (0)