如何在Keras中设置特定图层的偏置值

时间:2018-07-02 15:31:38

标签: keras

我试图在Keras中设置特定图层的偏差值,但没有找到任何方法。

例如,可以通过以下代码设置重量:

model.layers[-1].set_weights(weights)

有什么方法可以设置偏倚,可以设置权重的方法(如上)?

有人可以在这方面帮助我吗?

2 个答案:

答案 0 :(得分:1)

您可以为此使用K.set_value
   K.set_value(model.layers[-1].weights[1], np.ones((bias_dim,)))

请注意,当您使用set_weights时,必须提供一个既包含权重又包含偏差的列表,因此,如果您只想设置权重而不设置biais,则也可以使用K.set_value

答案 1 :(得分:0)

您可以通过为偏差或权重或两者设置约束来做到这一点。

在每一层都有bias_constraint和kernel_constraint,它们分别用于偏置和权重。 如果要将偏差值设置为 1,则示例如下。

model.add(Dense(1 ,bias_constraint=tf.keras.constraints.MinMaxNorm(
min_value=1.0, max_value=1.0, rate=1.0, axis=0), activation='softmax'))