
这次我们会介绍如何使用Tensorflow解决Classification(分类)问题。在之前的文章中介绍的都是Regression(回归)问题。分类和回归的区别在于输出变量的类型上。通俗理解定量输出是回归,或者说是连续变量预测;定性输出是分类,或者说是离散变量预测。如预测房价这是一个回归任务;把东西分成几类,比如猫狗猪牛,就是一个分类任务。
一、MNIST数据。
首先准备数据(MNIST库)
from tensorflow.examples.tutorials.mnist import input_data
# number 1 to 10 data
mnist = input_data.read_data_sets("MNIST_data",one_hot=True)
MNIST库是手写体数字库,差不多是这个样子的:

数据中包含55000张训练图片,每张图片的分辨率是28*28,所以我们的训练网络输入应该是28*28=784个像素数据。
二、搭建网络
(1)定义网络的输入
# define placeholder for inputs to network
xs =

本文介绍了使用Tensorflow解决分类问题的方法,以MNIST数据集为例,构建神经网络,包括数据准备、网络搭建、交叉熵损失函数和训练过程。简单网络在训练后表现出令人惊讶的图像识别精度。
最低0.47元/天 解锁文章
4903

被折叠的 条评论
为什么被折叠?



