tensorflow以及tflearn文档及案例

最新推荐文章于 2021-01-14 05:23:07 发布

辣大辣条

最新推荐文章于 2021-01-14 05:23:07 发布

阅读量436

点赞数

分类专栏：面经、算法题等

面经、算法题等专栏收录该内容

39 篇文章

订阅专栏

本文详细介绍使用TFLearn和TensorFlow进行IMDB电影评论情感分析的过程。从数据预处理到网络构建，再到训练模型，全面解析如何利用这些工具实现文本分类任务。通过实际案例展示TFLearn作为TensorFlow高级API的便捷性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

tflearn文档，比tensorflow更高一级的API： http://tflearn.org

tensorflow中文文档：http://www.tensorfly.cn/tfdoc/get_started/introduction.html

#-*-coding:utf-8-*-
from __future__ import division, print_function, absolute_import

import tflearn
from tflearn.data_utils import to_categorical, pad_sequences
from tflearn.datasets import imdb

# IMDB Dataset loading
train, test, _ = imdb.load_data(path='imdb.pkl', n_words=10000,
                                valid_portion=0.1)
trainX, trainY = train
testX, testY = test

# Data preprocessing
# Sequence padding
trainX = pad_sequences(trainX, maxlen=100, value=0.)
testX = pad_sequences(testX, maxlen=100, value=0.)
# print("trainX:",trainX)
# Converting labels to binary vectors
trainY = to_categorical(trainY,2)
testY = to_categorical(testY,2)
# print("trainY:",trainY)

# Network building
net = tflearn.input_data([None, 100])
# print("net1:",net)
# print(net.shape())
net = tflearn.embedding(net, input_dim=10000, output_dim=128)
# print("net2:",net)
# print(net.shape())
net = tflearn.lstm(net, 128, dropout=0.8)
net = tflearn.fully_connected(net, 2, activation='softmax')
net = tflearn.regression(net, optimizer='adam', learning_rate=0.001,
                         loss='categorical_crossentropy')

# Training
# model = tflearn.DNN(net, tensorboard_verbose=0)
# model.fit(trainX, trainY, validation_set=(testX, testY), show_metric=True,
#           batch_size=32)