lstm如何進行文本分類

Question

model = tf.keras.Sequential([
    encoder,
    tf.keras.layers.Embedding(len(encoder.get_vocabulary()), 64, mask_zero=True),
    tf.keras.layers.Bidirectional(tf.keras.layers.LSTM(64,  return_sequences=True)),
    tf.keras.layers.Bidirectional(tf.keras.layers.LSTM(32)),
    tf.keras.layers.Dense(64, activation='relu'),
    tf.keras.layers.Dropout(0.5),
    tf.keras.layers.Dense(1)

如何修改此代碼？

Answer 1

將tf.keras.layers.Bidirectional(tf.keras.layers.LSTM(64)),替換為tf.keras.layers.LSTM(64), 。

Answer 2

如果您將tensorflow-gpu與 Cudnn 一起使用，請使用

tf.keras.layers.CuDNNLSTM

它將改善訓練