当前位置:   article > 正文

使用LSTM网络实现文本情感分析_使用lstm进行文本情感分析

使用lstm进行文本情感分析

一、实验目的:

理解循环神经网络的基本概念和原理;了解循环神经网络处理文本数据的基本方法;掌握循环神经网络处理文本数据的实践方法,并实现文本情感分析任务。

  • 实验要求:

使用Keras框架定义并训练循环神经网络模型,并进行文本情感分析。

  1. import tensorflow as tf
  2. from tensorflow import keras
  3. import matplotlib.pyplot as plt
  4. import numpy as np
  5. # 加载 IMDB 数据
  6. imdb = keras.datasets.imdb
  7. (train_data, train_labels), (test_data, test_labels) = imdb.load_data(num_words=10000)
  8. print("训练记录数量:{},标签数量:{}".format(len(train_data), len(train_labels)))
  9. print(train_data[0])
  10. # 数据标准化
  11. train_data = keras.preprocessing.sequence.pad_sequences(train_data, padding='post', maxlen=256)
  12. test_data = keras.preprocessing.sequence.pad_sequences(test_data, padding='post', maxlen=256)
  13. print(train_data[0])
  14. # 构建模型
  15. vocab_size = 10000
  16. model = tf.keras.Sequential([tf.keras.layers.Embedding(vocab_size, 64),
  17. tf.keras.layers.Bidirectional(tf.keras.layers.LSTM(64)), tf.keras.
  18. layers.Dense(64, activation='relu'), tf.keras.layers.Dense(1)
  19. ])
  20. model.summary()
  21. # 配置并训练模型
  22. model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])
  23. x_val = train_data[:10000]
  24. partial_x_train = train_data[10000:]
  25. y_val = train_labels[:10000]
  26. partial_y_train = train_labels[10000:]
  27. history = model.fit(partial_x_train, partial_y_train, epochs=10, batch_size=512, validation_data=(x_val, y_val),
  28. verbose=1)
  29. result = model.evaluate(test_data, test_labels, verbose=2)
  30. print(result)
  31. # 训练过程可视化
  32. history_dict = history.history
  33. print(history_dict.keys())
  34. def plot_graphs(history, string):
  35. plt.plot(history.history[string])
  36. plt.plot(history.history['val_' + string])
  37. plt.xlabel("Epochs")
  38. plt.ylabel(string)
  39. plt.legend([string, 'val_' + string])
  40. plt.show()
  41. plot_graphs(history, "accuracy")
  42. plot_graphs(history, "loss")

运行结果可视化:

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小丑西瓜9/article/detail/734021
推荐阅读
相关标签
  

闽ICP备14008679号