keras从入门到放弃(十一）电影评价预测

keras基础实例

电影评价预测

导入数据集

import keras from keras import layers import matplotlib.pyplot as plt %matplotlib inline data = keras.datasets.imdb max_word = 10000 # 加载前10000个单词 最大不超过10000 (x_train, y_train), (x_test, y_test) = data.load_data(num_words=max_word)

查看数据

x_train.shape, y_train.shape OUT: ((25000,), (25000,)) x_train[0] y_train[0] OUT: 输出的是 词汇的index 输出的是 array([1, 0, 0, ..., 0, 1, 0], dtype=int64) 1 代表 正面评价 0 代表负面怕评价

加载index和词汇的对应关系

#加载词汇 word_index = data.get_word_index() #将index和value 互换 index_word = dict((value, key) for key,value in word_index.items())

打印第一个评

总结

以上是生活随笔为你收集整理的keras从入门到放弃(十一）电影评价预测的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。