首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >Tensorflow TextVectorization adapt() --检查生成的词汇表

Tensorflow TextVectorization adapt() --检查生成的词汇表
EN

Stack Overflow用户
提问于 2021-09-27 12:17:55
回答 1查看 1.1K关注 0票数 1

文本TextVectorization层用于文字编码,典型的工作流调用adapt()方法。

接下来,您将调用

以适应数据集预处理层的状态。这将导致模型生成字符串到整数的索引。

(https://www.tensorflow.org/tutorials/keras/text_classification)

如果需要,用户可以在dataset上调用该层的

()方法。当该层适应时,它将分析数据集,确定单个字符串值的频率,并从它们创建一个“词汇表”。

(https://www.tensorflow.org/api_docs/python/tf/keras/layers/TextVectorization#adapt)

adapt()操作的确切结果是什么,以及如何具体检查创建的词汇表的内容?

我的一小部分代码

代码语言:javascript
复制
seq_length = 100
vocab_size=50000

vectorize_layer = TextVectorization(
    max_tokens=vocab_size,
    output_mode='int',
    output_sequence_length=seq_length)

# build the vocabulary
vectorize_layer.adapt(text_ds)
EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-09-28 06:00:00

layer.get_vocabulary()这样做:

代码语言:javascript
复制
>>>data = tf.constant([["a", "c", "d"], ["d", "z", "b"]])
>>>layer = tf.keras.layers.StringLookup()
>>>layer.adapt(data)
>>>layer.get_vocabulary()

['[UNK]', 'd', 'z', 'c', 'b', 'a']

https://www.tensorflow.org/api_docs/python/tf/keras/layers/StringLookup

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/69346469

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档