老师，想看一下tensorflow2.x实现LSTM

来源：5-9 图像增强实战

慕慕65597133

2021-05-28

老师，想看一下tensorflow2.x实现lstm的代码

写回答

1回答

正十七

2021-06-07

tf2.x下实现lstm比较简单，tf2.x主推Keras的实现方式，例子如下：

embedding_dim = 16
batch_size = 512
single_rnn_model = keras.models.Sequential([
    # 1. define matrix: [vocab_size, embedding_dim]
    # 2. [1,2,3,4..], max_length * embedding_dim
    # 3. batch_size * max_length * embedding_dim
    keras.layers.Embedding(vocab_size, embedding_dim,
                           input_length = max_length),
    keras.layers.LSTM(units = 64, return_sequences = False),
    keras.layers.Dense(64, activation = 'relu'),
    keras.layers.Dense(1, activation='sigmoid'),
])

single_rnn_model.summary()
single_rnn_model.compile(optimizer = 'adam',
                         loss = 'binary_crossentropy',
                         metrics = ['accuracy'])

如果想学习tf2.x的编程，可以看我的另一门课程：https://coding.imooc.com/class/344.html

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学，深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

查看课程

相似问题

lstm实现，num_lstm_nodes与num_time_step含义理解、与实现x_size,h_size维度的理解

回答 1

老师可以出一下tensorflow2版本的代码

回答 1

老师可以解释下num_lstm_nodes的含义吗

回答 1

lstm cell实现的过程中h_size与h 含义的困惑？

回答 1

lstm的第一步输入

回答 3

打开慕课网App查看更多内容