【发布时间】:2018-10-10 16:05:02
【问题描述】:
我一直在关注 Towards Data Science 关于 word2vec 和 skip-gram 模型的教程,但我偶然发现了一个我无法解决的问题,尽管搜索了很多并尝试了多个不成功的解决方案。
它向您展示如何构建 skip-gram 模型架构的步骤似乎已被弃用,因为使用了 keras.layers 中的 Merge 层。
我试图做的是将他的代码(在 Keras 的 Sequential API 中实现)转换为功能 API,以解决 Merge 层的弃用问题,将其替换为 keras.layers.Dot 层。但是,我仍然停留在将两个模型(单词和上下文)合并为最终模型的这一步,其架构必须是这样的:
这是作者使用的代码:
from keras.layers import Merge
from keras.layers.core import Dense, Reshape
from keras.layers.embeddings import Embedding
from keras.models import Sequential
# build skip-gram architecture
word_model = Sequential()
word_model.add(Embedding(vocab_size, embed_size,
embeddings_initializer="glorot_uniform",
input_length=1))
word_model.add(Reshape((embed_size, )))
context_model = Sequential()
context_model.add(Embedding(vocab_size, embed_size,
embeddings_initializer="glorot_uniform",
input_length=1))
context_model.add(Reshape((embed_size,)))
model = Sequential()
model.add(Merge([word_model, context_model], mode="dot"))
model.add(Dense(1, kernel_initializer="glorot_uniform", activation="sigmoid"))
model.compile(loss="mean_squared_error", optimizer="rmsprop")
这是我将顺序代码实现转换为功能代码实现的尝试:
from keras import models
from keras import layers
from keras import Input, Model
word_input = Input(shape=(1,))
word_x = layers.Embedding(vocab_size, embed_size, embeddings_initializer='glorot_uniform')(word_input)
word_reshape = layers.Reshape((embed_size,))(word_x)
word_model = Model(word_input, word_reshape)
context_input = Input(shape=(1,))
context_x = layers.Embedding(vocab_size, embed_size, embeddings_initializer='glorot_uniform')(context_input)
context_reshape = layers.Reshape((embed_size,))(context_x)
context_model = Model(context_input, context_reshape)
model_input = layers.dot([word_model, context_model], axes=1, normalize=False)
model_output = layers.Dense(1, kernel_initializer='glorot_uniform', activation='sigmoid')
model = Model(model_input, model_output)
但是,执行时,返回如下错误:
ValueError: Layer dot_5 被调用时输入不是符号 张量。接收类型:.满的 输入: [, ]。所有输入 该层应该是张量。
我是 Keras 函数式 API 的初学者,如果你能在这种情况下给我一些指导,我将如何将上下文和单词模型输入到点层以实现架构图片。
【问题讨论】:
标签: python tensorflow keras word2vec word-embedding