今天我们将使用tensorflow自带的手写数字数据,来训练一个softmax regression的模型。这应该是机器学习领域的入门级任务。这些数据TensorFlow已经做了相当完整的包装,我们可以先忽略数据上的细节,把重点放到算法和结构上来。
from tensorflow.examples.tutorials.mnist import input_data mnist=input_data.read_data_sets('MNIST_data',one_hot=True) print(mnist.train.images.shape,mnist.train.labels.shape) #训练集的数量,维度,类别信息 print(mnist.test.images.shape,m