我想在 gensim 中仅加载一个文件的扩展名。
正常的代码是这样的:
model = gensim.models.word2vec.Word2Vec.load("news.bin")
但我希望它自动打开带有“.bin”的任何文件。
例子:
model = gensim.models.word2vec.Word2Vec.load(***I would like to change this part to only load any .bin***)
.bin 文件:
它可以是“news.bin”、“file.bin”或“guess.bin”。只要它只加载扩展。谢谢你。
如果你想打开全部由其中一人一个,你可以将文件遍历在目标目录中。
这是 Python 3 的代码示例:
import os
directory_path = "/path/to/directory"
for filename in os.listdir(directory_path):
if filename.endswith(".bin"):
file_path = os.path.join(directory_path, filename)
model = gensim.models.word2vec.Word2Vec.load(file_path)
# Do whatever you want to do with model
如果您只想打开其中任何一个,您可以在第一场比赛后跳出 for 循环:
import os
directory_path = "/path/to/directory"
for filename in os.listdir(directory_path):
if filename.endswith(".bin"):
file_path = os.path.join(directory_path, filename)
model = gensim.models.word2vec.Word2Vec.load(file_path)
# Do whatever you want to do with model
# Break out of the for loop afterwards so it stops iterating
break
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句