本文共 691 字，大约阅读时间需要 2 分钟。

回顾以前的笔记

在keras中，内置了imdb电影评分数据集，来进行评价预测

安装keras

conda install keras

conda就帮依赖全部搞定，记得加源

导入imdb

from keras.datasets import imdb

数据集简要说明

• 一个长长的英文句子，有的有几千单词，有的有几十，分类成好的评价和不好的评价

在数据中不是单词，而是单词的索引

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

查看数据

x_train.shape, y_train.shapeOUT:((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

转载地址：https://maoli.blog.csdn.net/article/details/102257429 如侵犯您的版权，请留言回复原文章的地址，我们会给您删除此文章，给您带来不便请您谅解！