首页 > 解决方案 > MLP 中输出层的默认权重是多少?

问题描述

我写了两个版本的程序,一个是在最后一个输出层有权重,一个是没有设置。如果在隐藏层和输出层中没有明确提及,您能否建议权重程序?

代码片段 - 版本 1

model = Sequential()
model.add(Dense(units = 6, kernel_initializer = "GlorotUniform",activation='relu',input_dim = 6))     #First layer
model.add(Dense(units = 6, kernel_initializer = "GlorotUniform",activation='relu'))  # Second hidden layer
model.add(Dense(units = 1,kernel_initializer = "GlorotUniform"))   # Output layer 

代码片段 - 版本 2

model = Sequential()
model.add(Dense(units = 6, kernel_initializer = "GlorotUniform",activation='relu',input_dim = 6))     #First layer
model.add(Dense(units = 6, kernel_initializer = "GlorotUniform",activation='relu'))  # Second hidden layer
model.add(Dense(units = 1))   # Output layer 

标签: tensorflowkeras

解决方案


TensorFlow Dense 层中的默认权重初始化器是glorot_uniform
因此,即使您没有明确提及它,它也会使用glorot_uniform

在此处查看文档


推荐阅读