【发布时间】:2021-05-07 15:05:30
【问题描述】:
是否可以在 Keras 中构建具有 3 个输出密集层(每个具有 10 个神经元)的网络?
如果可以,您能给我一些建议吗?
【问题讨论】:
-
好的,看看Functional API。
标签: keras neural-network
是否可以在 Keras 中构建具有 3 个输出密集层(每个具有 10 个神经元)的网络?
如果可以,您能给我一些建议吗?
【问题讨论】:
标签: keras neural-network
例如,像这样的:
from tensorflow.keras.layers import Input, Dense
from tensorflow.keras.models import Model
input_layer = Input(shape=(20,))
shared_layer = Dense(20)(input_layer)
output1 = Dense(10)(shared_layer)
output2 = Dense(10)(shared_layer)
output3 = Dense(10)(shared_layer)
model = Model(inputs=input_layer, outputs=[output1, output2, output3])
print(model.summary())
import numpy as np
a = np.random.randn(10,20)
b1 = np.random.rand(10,10)
b2 = np.random.rand(10,10)
b3 = np.random.rand(10,10)
model.compile(optimiser='adam', loss='mean_squared_error')
model.fit(a, [b1,b2,b3], epochs=10)
(随机数据)
您可能需要调整损失函数,具体取决于您的实际目标。
【讨论】: