tensorflow学习笔记（MNIST-2）

# -*- coding:utf-8 -*-
# "-*- coding:utf-8 -*-" 定义源代码的编码格式，#
from tensorflow.python.platform import gfile
from tensorflow.contrib.learn.python.learn.datasets.mnist import extract_images
from tensorflow.contrib.learn.python.learn.datasets.mnist import extract_labels
from tensorflow.contrib.learn.python.learn.datasets.mnist import DataSet
from tensorflow.contrib.learn.python.learn.datasets import base

import numpy as np
import pandas as pd
import tensorflow as tf

def input_data_from_local():
    
    # train_x,train_y,test_x,test_y的文件路径
    TRAIN_IMAGES = 'train-images-idx3-ubyte.gz'
    TRAIN_LABELS = 'train-labels-idx1-ubyte.gz'
    TEST_IMAGES = 't10k-images-idx3-ubyte.gz'
    TEST_LABELS = 't10k-labels-idx1-ubyte.gz'
    
    # 打开文件使用with操作，使用结束后自动关闭文件
    with gfile.Open(TRAIN_IMAGES,'rb') as f:
        train_images=extract_images(f)
    #print (train_images.shape)
    
    # 读取labels时one-hot=True,可以将[1,2,3]转化为[[0,1,0,0,0,0,0,0,0,0],[0,0,1,0,0,0,0,0,0,0,0],[0,0,0,1,0,0,0,0,0,0]]
    with gfile.Open(TRAIN_LABELS,'rb') as f:
        train_labels=extract_labels(f,one_hot=True)
    #print (train_labels.shape)
    
    with gfile.Open(TEST_IMAGES,'rb') as f:
        test_images=extract_images(f)
    #print (test_images.shape)
    
    with gfile.Open(TEST_LABELS,'rb') as f