LSTM百度百科
- 长短期记忆网络(LSTM,Long Short-Term Memory)是一种时间循环神经网络,是为了解决一般的RNN(循环神经网络)存在的长期依赖问题而专门设计出来的,所有的RNN都具有一种重复神经网络模块的链式形式。在标准RNN中,这个重复的结构模块只有一个非常简单的结构,例如一个tanh层。
学习目标:
了解LSTM在自然语言处理中的使用
代码:
from keras.datasets import imdb
# 导入模型
from keras import Sequential
from keras.layers import Embedding, LSTM, Dense, Dropout
vocalbulary_size = 5000 # 词汇量
(x_train, y_train), (x_test, y_test) = imdb.load_data(num_words=vocalbulary_size)
print('Loaded dataset with {} training samples, {} test samples'.format(len(x_train), len(x_test))