Tensorflow 基于minst手写数字数据集合建立逻辑回归分类模型

THE MNIST DATABASE of handwritten digits（点击进入下载4个数据文件）

加载一下数据集

import numpy as np
import tensorflow as tf
import matplotlib.pyplot as plt
import input_data
import warnings
warnings.filterwarnings("ignore")
mnist      = input_data.read_data_sets('data/', one_hot=True)#进行度热编码
trainimg   = mnist.train.images
trainlabel = mnist.train.labels
testimg    = mnist.test.images
testlabel  = mnist.test.labels
print ("MNIST loaded")

print (trainimg.shape)
print (trainlabel.shape)
print (testimg.shape)
print (testlabel.shape)
#print (trainimg)
print (trainlabel[0])