python - 如何训练具有 2D 输出的网络？ ( python ，凯拉斯)

Question

我想训练一个回归网络，其输出是两个坐标(x1，y1)和(x2，y2)。我的问题是:如果我想训练网络，我的输出应该分开吗？我的意思是我的输出应该是这样的:

 [x1,y1,x2,y2] 

或者有没有办法将它们堆叠起来，例如:

[(x1,y1),(x2,y2)]

提前致谢

Answer 1

RepeatVector 就是用于此目的(请参阅 Keras 文档)。

您希望输出形状为 (2, 2)，或者两个坐标的数组，每个坐标有两个条目。

num_outputs = 2
num_elements = 2

添加用于处理网络输入的层后，添加一个RepeatVector。这将使获得数据数组作为输出成为可能。您的最后一层需要输出 num_elements 例如:

model.add(RepeatVector(num_outputs))
# Optional layers can be inserted here
model.add(Dense(num_elements))

简而言之，重复向量复制前一层的内容以发送到进一步的节点，每个路径将输出一个 num_elements 数组，为您提供所需的输出形状。请注意，您的训练数据(标签)也必须由形状 (2, 2) 数组组成。