今天寫個程序想用Merge層實現兩個模型合併輸出,可是模型在訓練的時候一直出錯,爲了解決問題,又把官方文檔好好看了下,找了些資料。node
首先keras的文檔中是這樣給出的,把若干個層合併成一個層網絡
keras.engine.topology.Merge(layers=None, mode='sum', concat_axis=-1, dot_axes=-1, output_shape=None, node_indices=None, tensor_indices=None, name=None)
layers:該參數爲Keras張量的列表,或Keras層對象的列表。該列表的元素數目必須大於1。spa
mode:合併模式,若是爲字符串,則爲下列值之一{「sum」,「mul」,「concat」,「ave」,「cos」,「dot」}code
其中sum和mul是對待合併層輸出作一個簡單的求和、乘積運算,所以要求待合併層輸出shape要一致。concat是將待合併層輸出沿着最後一個維度進行拼接,所以要求待合併層輸出只有最後一個維度不一樣。 對象
下面我要說的主要針對訓練時,Merge是一個層對象,在多個sequential組成的網絡模型中,若是blog
x:輸入數據。若是模型只有一個輸入,那麼x的類型是numpy array,若是模型有多個輸入,那麼x的類型應當爲list,list的元素是對應於各個輸入的numpy array文檔
y:標籤,numpy array字符串
不然運行時極可能會提示意思就是你輸入的維度與實際不符,好了,說什麼都不如寫一個代碼來的實際:input
model_left= Sequential() model_left.add(Dense(50, input_shape=(784,))) model_left.add(Activation('relu')) model_right = Sequential() model_right.add(Dense(50, input_shape=(784,))) model_rightadd(Activation('relu')) model = Sequential() model.add(Merge([model_left,model_right], mode='concat')) model.add(Dense(10)) model.add(Activation('softmax')) model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy']) model.fit([X_train, X_train], Y_train, batch_size=64, nb_epoch=30, validation_data=([X_test, X_test], Y_test))