我正在开发一个前馈神经网络,使用Keras嵌入技术。
我想为嵌入层设置偏置权重,但不确定如何实现。
Keras Dense层允许指定use_bias=True,然后设置偏置权重。 对于Embedding层是否有相应的方法呢?
Keras Dense层允许指定use_bias=True,然后设置偏置权重。 对于Embedding层是否有相应的方法呢?
from keras.models import Model
from keras.layers import Embedding, Input, Add, Dot
a = Input(shape=(1,))
b = Input(shape=(1,))
emb_a = Embedding(num_words+1, 50)(a)
bias_a = Embedding(num_words+1, 1)(a)
emb_b = Embedding(num_words+1, 50)(b)
bias_b = Embedding(num_words+1, 1)(b)
dot = Dot(axes=-1)([a,b])
add = Add()([dot,bias_a,bias_b])