Menggabungkan dua model berbeda dalam Keras

Saya mencoba menggabungkan dua model Keras menjadi satu model dan saya tidak dapat melakukannya.

Sebagai contoh pada Gambar terlampir, saya ingin mengambil lapisan tengah dari dimensi 8, dan menggunakan ini sebagai input ke lapisan (dari dimensi 8 lagi) di Model dan kemudian menggabungkan kedua Model dan Model sebagai model tunggal.$A2$$B1$$B$$A$$B$

Saya menggunakan modul fungsional untuk membuat Model dan Model secara mandiri. Bagaimana saya bisa menyelesaikan tugas ini?$A$$B$

Catatan : adalah masukan lapisan model dan adalah masukan lapisan model .$A1$$A$$B1$$B$

Saya menemukan jawaban untuk pertanyaan saya dan di sini adalah kode yang dibangun di atas jawaban di atas.

from keras.layers import Input, Dense
from keras.models import Model
from keras.utils import plot_model

A1 = Input(shape=(30,),name='A1')
A2 = Dense(8, activation='relu',name='A2')(A1)
A3 = Dense(30, activation='relu',name='A3')(A2)

B2 = Dense(40, activation='relu',name='B2')(A2)
B3 = Dense(30, activation='relu',name='B3')(B2)

merged = Model(inputs=[A1],outputs=[A3,B3])
plot_model(merged,to_file='demo.png',show_shapes=True)

dan di sini adalah struktur output yang saya inginkan:

Di Keras ada cara yang bermanfaat untuk mendefinisikan model: menggunakan API fungsional . Dengan API fungsional, Anda dapat menentukan grafik lapisan asiklik yang diarahkan, yang memungkinkan Anda membangun arsitektur yang sepenuhnya arbitrer. Mempertimbangkan contoh Anda:

#A_data = np.zeros((1,30))
#A_labels = np.zeros((1,30))
#B_labels =np.zeros((1,30))

A1 = layers.Input(shape=(30,), name='A_input')
A2 = layers.Dense(8, activation='???')(A1)
A3 = layers.Dense(30, activation='???', name='A_output')(A2)


B2 = layers.Dense(40, activation='???')(A2)
B3 = layers.Dense(30, activation='???', name='B_output')(B2)

## define A
A = models.Model(inputs=A1, outputs=A3)

## define B
B = models.Model(inputs=A1, outputs=B3) 

B.compile(optimizer='??',
          loss={'B_output': '??'}
          )

B.fit({'A_input': A_data},
  {'B_output': B_labels},
  epochs=??, batch_size=??)

Jadi begitulah! Anda dapat melihat hasilnya dengan B.summary()::

Layer (type)                 Output Shape              Param    
A_input (InputLayer)         (None, 30)                0         
_________________________________________________________________
dense_8 (Dense)              (None, 8)                 248     
______________________________________________________________
dense_9 (Dense)              (None, 40)                360       
_________________________________________________________________
B_output (Dense)             (None, 30)                1230