Skip to content

Commit 0d629af

Browse files
committed
add vocab size
1 parent f8bfd3e commit 0d629af

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)