bheinzerling / bpemb

Pre-trained subword embeddings in 275 languages, based on Byte-Pair Encoding (BPE)
https://nlp.h-its.org/bpemb
MIT License
1.18k stars 101 forks source link

Load custom Word2Vec #48

Closed Delphine22 closed 4 years ago

Delphine22 commented 4 years ago

Thanks for this very useful package. It seems loading custom Word2Vec embeddings is not possible. With KeyedVectors function, I can't load my custom Word2Vec model (trained on custom data). I know I can only load the model.bin with Word2Vec.load function. Is it possible to add this similar function in order to enable custom model loading?

bheinzerling commented 4 years ago

Yes, there is support for loading custom models:

https://github.com/bheinzerling/bpemb/issues/38#issuecomment-604217177