MNIST神经网络python实现

import os
import struct
import numpy as np
import matplotlib.pyplot as plt
import gzip
import tempfile
import random
import tensorflow as tf
#*****************************************************************************************
#从文件中读入数据
def load_mnist(labels_path,images_path):
    """Load MNIST data from `path`"""
    with open(labels_path, 'rb') as lbpath:
        magic, n = struct.unpack('>II',
                                 lbpath.read(8))
        labels = np.fromfile(lbpath,
                             dtype=np.uint8)
    with open(images_path, 'rb') as imgpath:
        magic, num, rows, cols = struct.unpack('>IIII',
                                               imgpath.read(16))
        images = np.fromfile(imgpath,
                             dtype=np.uint8).reshape(len(labels), 784)
    return images, labels
train_labels_path = 'F:\Dataset\MNIST\MNIST/train-labels.idx1-ubyte'
train_images_path = 'F:\Dataset\MNIST\MNIST/train-images.idx3-ubyte'
test_labels_path = 'F:\Dataset\MNIST\M