首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >神经网络启动权值的单独设置

神经网络启动权值的单独设置
EN

Stack Overflow用户
提问于 2021-04-13 16:38:23
回答 1查看 106关注 0票数 0

我有一个简单的前馈神经网络,由8个输入神经元组成,然后是2个隐层,每个隐层有6个隐神经元,1个输出层由1个输出神经元组成。

Keras代码是:

代码语言:javascript
复制
model = Sequential()

model.add(Dense(6, input_dim = 8, activation='tanh')
model.add(Dense(6, activation='tanh'))
model.add(Dense(1, activation='tanh'))

问题:

由于我知道8个输入参数中哪一个对单个输出的影响最大,所以我可以将它们的起始权值设置为相对于其他输入参数的更高的值。如果这是可能的,这可以大大减少培训时间(如果我没有错)。

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-04-13 20:13:32

代码语言:javascript
复制
# reading the initial weights and bias of the input layer
layer_1 = (model.layers)[0]

# reading the initial weights of the input layer
w_1 = layer_1.get_weights()[0]

# setting weights for nth parameter of the input layer to a modified value val
w_1[n, :] = val

# setting the modified weights and unmodified bias of the input layer 
layer_1.set_weights([w_1, layer_1.get_weights()[1]])

# writing layer_1 to model
(model.layers)[0] = layer_1
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/67078973

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档