tensorflow——lmdb

最新推荐文章于 2020-11-08 22:24:19 发布

Coco~567

最新推荐文章于 2020-11-08 22:24:19 发布

阅读量864

点赞数 1

分类专栏： tensorflow 文章标签： imdb

本文链接：https://blog.youkuaiyun.com/u012198575/article/details/97370663

版权

参考官网：https://www.tensorflow.org/tutorials/keras/basic_text_classification

与源码不同，设置index_from=0从第0位开始，实际上map类型的word_index会从1开始。

import tensorflow as tf
from tensorflow import keras
import numpy as np

imdb = keras.datasets.imdb
(x_train, y_train), (x_test, y_test) = keras.datasets.imdb.load_data(num_words=10000, index_from=0)

print("train len >> %d, test len >> %d" %(len(x_train), len(x_test)))

print(x_train[0]) #每个整数表示字典中的一个特定字词

#将整数转换回字词
# A dictionary mapping words to an integer index
word_index = imdb.get_word_index()

# The first indices are reserved
#word_index = {k:(v+3) for k,v in word_index.item