【发布时间】:2020-03-26 01:15:23
【问题描述】:
我正在关注 tensorflow 网站 (https://www.tensorflow.org/tutorials/text/word_embeddings#create_a_simple_model) 上的教程来学习词嵌入,我的困惑是在嵌入层之后有一个 Globalaveragepooling 层的目的,如下所示:
model = keras.Sequential([
layers.Embedding(encoder.vocab_size, embedding_dim),
layers.GlobalAveragePooling1D(),
layers.Dense(16, activation='relu'),
layers.Dense(1)
])
我了解池的含义以及它是如何完成的。如果有人能解释为什么我们需要一个池化层,以及如果我们不使用它会发生什么变化,我将不胜感激。
【问题讨论】:
标签: tensorflow machine-learning keras deep-learning nlp