обучение двух сетей одновременно в тензорном потоке

Question

Представьте себе x_1 и x_2 - два входа нейронной сети, а y_1 и y_2 - выходы. Затем y_1 и y_2 должны быть объединены и переданы во вторую сеть. Выход второй сети представляет собой одну двоичную метку z={0,1}. У меня есть матрица X=[x_1;x_2;...;x_2n] (вход первой сети) и метки Z = [z_1;z_2;...;z_n] (выход второй сети).

Как настроить эту структуру в тензорном потоке?

Answer 1

Предполагая, что у вас есть две модели с именами model1 и model2, а ваши входы x_1 и x_2 являются тензорными входами для model1:

x = [x_1, x_2]
y = model1(x)
y = tf.keras.layers.Concatenate()(y)
z = model2(y)
combined_model = tf.keras.Model(x, z)