word2vec模型如何使用
Created by: stopit
想直接利用hub.Module(name="word2vec_skipgram")拿到word2vec的模型直接使用,但是没找到demo。能否补充一下这块。
现在的测试代码是这样的,但是运行会报错:
import paddlehub as hub
module = hub.Module(name="word2vec_skipgram") inputs, outputs, program = module.context(trainable=False)
word_ids = inputs["word_ids"] embedding = outputs["word_embs"] print(embedding[1123])