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

132 stars 12 forks source link

Fixed import of .bin files in word2vec #14

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.

pdrm83 commented 2 years ago

Great job! Thanks.

almarengo commented 2 years ago

Great! Thank you.