tensorflow逻辑回归

本文利用tensorflow构建逻辑回归模型，利用softmax分类器对mnist手写字体进行识别分类。每个样本是28*28大小的。

一、导入数据集

import tensorflow as tf
import numpy as np
from tensorflow.examples.tutorials.mnist import input_data

#读入手写字体数据集，会下载tensorFlow自带的mnist手写字体数据集
mnist = input_data.read_data_sets('MNIST_data',one_hot=True)

trainimg = mnist.train.images
trainlabel = mnist.train.labels
testimg    = mnist.test.images
testlabel  = mnist.test.labels

#简单看一下样本的结构
print (trainimg.shape)
print (trainlabel.shape)
print (testimg.shape)
print (testlabel.shape)
print (trainlabel[0])

####运行结果如下：
Extracting data\train-images-idx3-ubyte.gz
Extracting data\train-labels-idx1-ubyte.gz
Extracting data\t10k-images-idx3-ubyte.gz
Extracting data\t10k-labels-idx1-u