我在Keras之上阅读了一些IMDB电影评论情绪分类器。下面是模型定义:
model = keras.Sequential()
model.add(keras.layers.Embedding(vocab_size, 16))
model.add(keras.layers.GlobalAveragePooling1D())
model.add(keras.layers.Dense(16, activation="relu"))
model.add(keras.layers.Dense(1, activation="sigmoid"))
我不明白的是GlobalAveragePooling1D在这里的作用。
为了具体回答它为什么存在(而不是它是如何工作的(,model.summary()
将揭示它正在提供降维。