Marge two models in tensorflow

已查看 17 次
跳至第一个未读帖子

Vivek Kumar

未读,
2022年11月2日 06:38:312022/11/2
收件人 TensorFlow Hub
Hi All,
       how to two models concat or marge model1 & model2 both are written in TensorFlow. Please help me with this issue.

-------------------------Model 2 --> RNN Model---------------------------------


model1 = tf.keras.Sequential([
tf.keras.layers.Embedding(vocab_size, embedding_dim, input_length = max_length),
tf.keras.layers.Bidirectional(tf.keras.layers.LSTM(32)),
tf.keras.layers.Dense(1, activation='relu'),
tf.keras.layers.Dense(1, activation='sigmoid')
])

model1.compile(loss='binary_crossentropy',optimizer='adam',metrics=['accuracy'])
# history1 = model1.fit(training_padded, training_labels, epochs=2, validation_data=(testing_padded, testing_labels), verbose=2)
score,acc=model1.evaluate(testing_padded,testing_labels)
acc,score

#RNN Model
print('''

-------------------------Model 2 --> RNN Model---------------------------------

''')
model2 = tf.keras.Sequential([
tf.keras.layers.Embedding(vocab_size, embedding_dim, input_length=max_length),
tf.keras.layers.Bidirectional(tf.keras.layers.SimpleRNN(50, return_sequences=True,input_shape=(max_length, embedding_dim))),
tf.keras.layers.Dropout(0.5),
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(1, activation='sigmoid')
])

model2.compile(loss='binary_crossentropy',optimizer='adam',metrics=['accuracy'])
history2 = model2.fit(training_padded, training_labels, epochs=2, validation_data=(testing_padded, testing_labels), verbose=2)
score,acc=model2.evaluate(testing_padded,testing_labels)
acc,score


Vivek Kumar

未读,
2022年11月2日 06:41:502022/11/2
收件人 TensorFlow Hub、Vivek Kumar
LSTM and RNN Models
回复全部
回复作者
转发
0 个新帖子