加载imdb数据以及TensorBoard的使用方法

最新推荐文章于 2025-10-21 19:09:17 发布

原创

最新推荐文章于 2025-10-21 19:09:17 发布 · 968 阅读

0 ·

CC 4.0 BY-SA版权

加载imdb数据

修改imdb.py

这里我把imdb数据下载下来，放在keras>datasets文件夹下面，并且需要修改一下imdb.py文件

# path = get_file(path,
# origin='https://s3.amazonaws.com/text-datasets/imdb.npz',
# file_hash='599dadb1135973df5b59232a0e9a887c')
    path = r'D:\pythonText\venv\Lib\site-packages\keras\datasets\imdb.npz'

这里把原来使用的path路径注释掉，使用我们用来放置imdb.npz的路径

加载imdb数据

from keras.datasets import imdb
from keras import preprocessing

max_feature = 10000
max_len = 100

(train_data, train_labels), (test_data, test_labels) = imdb.load_data(num_words=max_feature)
train_data = preprocessing.sequence.pad_sequences