text: def load_mnist(path, kind='train'):
      import os
      import gzip
      import numpy as np
  
      """Load MNIST data from `path`"""
      labels_path = os.path.join(path,
                                 '%s-labels-idx1-ubyte.gz'
                                 % kind)
      images_path = os.path.join(path,
                                 '%s-images-idx3-ubyte.gz'
                                 % kind)
  
      with gzip.open(labels_path, 'rb') as lbpath:
          labels = np.frombuffer(lbpath.read(), dtype=np.uint8,
                                 offset=8)
  
      with gzip.open(images_path, 'rb') as imgpath:
          images = np.frombuffer(imgpath.read(), dtype=np.uint8,
                                 offset=16).reshape(len(labels), 4096)
  
      return images, labels
  
  
  