Tensorflow tflearn

1、定义模型import tflearn
from tflearn.layers.core import input_data, dropout, fully_connected
from tflearn.layers.conv import conv_2d, max_pool_2d
from tflearn.layers.estimator import regressionimport tflearn.datasets.mnist as mnisttrainX, trainY, testX, testY = mnist.load_data(data_dir="../../datasets/MNIST_data", one_hot=True)
# 将图像数据resize成卷积卷积神经网络输入的格式。
trainX = trainX.reshape([-1, 28, 28, 1])
testX = testX.reshape([-1, 28, 28, 1])# 构建神经网络。
net = input_data(shape=[None, 28, 28, 1], name='input')
net = conv_2d(net, 32, 5, activation='relu')
net = max_pool_2d(net, 2)
net = conv_2d(net, 64, 5, activation='relu')
net = max_pool_2d(net, 2)
net = fully_connected(net, 500, activation='relu')
net = fully_connected(net, 10, activation='softmax')
# 定义学习任务。指定优化器为sgd，学习率为0.01，损失函数为交叉熵。
net = regression(net, optimizer='sgd', learning_rate=0.01,loss='categorical_crossentropy')

2、训练模型

# 通过定义的网络结构训练模型，并在指定的验证数据上验证模型的效果。
model = tflearn.DNN(net, tensorboard_verbose=0)
model.fit(trainX, trainY, n_epoch=10,validation_set=([testX, testY]),show_metric=True)

Tensorflow tflearn相关推荐

学习笔记CB013: TensorFlow、TensorBoard、seq2seq
2019独角兽企业重金招聘Python工程师标准>>> tensorflow基于图结构深度学习框架,内部通过session实现图和计算内核交互. tensorflow基本数学运算用法 ...
TensorFlow工具及笔记
机器学习工具安装# pip3 install tensorflow tflearn pandas matplotlib numpy scipy scikit-learn模型:是一个一元二次函数方程:f ...
用 TensorFlow 做个聊天机器人
上一次提到了不错的学习聊天机器人的资源,不知道小伙伴们有没有去学习呢. 自己动手做聊天机器人教程我最近每天都会学一点,拿出解读来和大家分享一下. 本文结构: 聊天机器人的架构简图用 TensorF ...
TensorFlow 1.x 深度学习秘籍：1~5
原文:TensorFlow 1.x Deep Learning Cookbook 协议:CC BY-NC-SA 4.0 译者:飞龙本文来自[ApacheCN 深度学习译文集],采用译后编辑(MTP ...
在TensorFlow中使用深度学习GANs处理图像
设计师和摄影师用内容自动填补来补充图像中不想要的或缺失的部分.与之相似的技术还有图像完善和修复.实现内容自动填补,图像完善和修复的方法有很多.本文介绍的是 Raymond Yeh 和 Chen Che ...
深度学习（十五）——SPPNet, Fast R-CNN
https://antkillerfarm.github.io/ RCNN(续) RCNN算法的基本流程 RCNN算法分为4个步骤: Step 1:候选区域生成.一张图像生成1K~2K个候选区域(采用 ...
python 怎么得到图像深度图软件_如何用 Python 和 fast.ai 做图像深度迁移学习？...
本文带你认识一个优秀的新深度学习框架,了解深度学习中最重要的3件事. 框架看到这个题目,你可能会疑惑:老师,你不是讲过如何用深度学习做图像分类了吗?迁移学习好像也讲过了啊! 说得对!我要感谢你对我专 ...
第955期机器学习日报(2017-04-30)
机器学习日报 2017-04-30 科普:Amazon 首席执行官 Jeff Bezos "人人都应该知道的人工智能知识" @新智元用于性能分析.模型优化的神经网络生成器 @爱可 ...
Tensorflow【实战Google深度学习框架】使用 TFLearn 实现逻辑运算符
文章目录导库逻辑非(NOT)运算符逻辑或(OR)运算符逻辑与(AND)运算符逻辑异或(XOR)运算符导库通过使用TFLearn实现简单的逻辑非(NOT).逻辑或(OR).逻辑与(AND) ...

Tensorflow tflearn

Tensorflow tflearn相关推荐

最新文章

热门文章