Fixed import of bin files in word2vec

pdrm83 / sent2vec

How to encode sentences in a high-dimensional vector space, a.k.a., sentence embedding.

MIT License

133 stars 12 forks source link

Fixed import of bin files in word2vec #13

Closed almarengo closed 2 years ago

almarengo commented 2 years ago

This PR will fix the problem when loading Google news 300 model into word2vec. The original code was not supporting binary file for pretrained vectors. Now the code will check if the the file is .bin or .txt and will pass the right argument.