Keras合并层警告

时间:2017-04-01 17:53:59

标签: python deep-learning keras

我一直在收到这个警告:

  

lstm.py:119:UserWarning:merge函数已弃用,将于08/2017之后删除。请使用keras.layers.merge中的图层,例如addconcatenate等     merged_vector = merge([l1,l2],mode = lambda x:(x [0] - x [1])** 2,output_shape = lambda x:x [0])   /Library/Python/2.7/site-packages/keras/legacy/layers.py:456:UserWarning:Merge图层已弃用,将于08/2017之后删除。请使用keras.layers.merge中的图层,例如addconcatenate等     名称=名)

与以下代码行有关:

merged_vector = merge([l1, l2], mode=lambda x: (x[0] - x[1])**2, output_shape=lambda x: x[0])

我的模型运行正常,但如何在Keras 2.0.2中实现自定义合并?谢谢。

1 个答案:

答案 0 :(得分:9)

回答我自己的问题:

# Custom Merge
def euclid_dist(v):
    return (v[0] - v[1])**2

def out_shape(shapes):
    return shapes[0]

merged_vector = Lambda(euclid_dist, output_shape=out_shape)([l1, l2])