我想训练一个回归网络,其输出是两个坐标(x1,y1)和(x2,y2)。 我的问题是:如果我想训练网络,我的输出应该分开吗? 我的意思是我的输出应该是这样的:
[x1,y1,x2,y2]
或者有没有办法将它们堆叠起来,例如:
[(x1,y1),(x2,y2)]
提前致谢
最佳答案
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)
数组组成。
关于python - 如何训练具有 2D 输出的网络? ( python ,凯拉斯),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41006159/