我希望使用一个变量,比如已经为 gensim 训练了某些句子的模型。例如,我使用 gensim word2vec 训练一个句子在另一个函数中找到它的向量并将其保存到一个名为“model”的变量中。
然后我创建另一个函数来获取每个单词的向量并返回它。
我的代码如下:
def gen(sentence):
model = gensim.models.word2vec.Word2Vec([sentence],min_count=1, workers=1, size=3)
....
....
return ...
def name(sentence):
for word in sentence:
print(model.wv[word])
return
有没有办法将变量从名为“model”的函数 def “gen” 传递给函数 def “name”?我尝试使用 self 但它不起作用。我只想调用该变量,因为我不想构建另一个额外的函数来链接它。
相关分类