【发布时间】:2021-04-01 15:36:57
【问题描述】:
我现在使用这个https://tfhub.dev/google/Wiki-words-500/2 作为预训练嵌入。对于我的应用程序(文本生成),我有兴趣获取词汇表中的标记列表。我知道它的尺寸是 1009375。但我似乎无法获得清单。我知道它已加载到内存中,但我似乎找不到它。有人知道怎么做吗?
【问题讨论】:
标签: tensorflow tensorflow-datasets
我现在使用这个https://tfhub.dev/google/Wiki-words-500/2 作为预训练嵌入。对于我的应用程序(文本生成),我有兴趣获取词汇表中的标记列表。我知道它的尺寸是 1009375。但我似乎无法获得清单。我知道它已加载到内存中,但我似乎找不到它。有人知道怎么做吗?
【问题讨论】:
标签: tensorflow tensorflow-datasets
google/Wiki-words-500/2 模型的默认缓存位置是 /tmp/tfhub_modules/bf115a5fe517f019bebae05b433eaeee6415f5bf (more on caching)。您将在 assets 目录中找到该文件,其中包含辅助文件:
wc -l /tmp/tfhub_modules/bf115a5fe517f019bebae05b433eaeee6415f5bf/assets/tokens.txt
> 1009374
【讨论】: