code: def load_data():
      X, y = datasets.fetch_openml(
          "mnist_784", version=1, return_X_y=True
      )
  
      X = X / 255.0
      X -= X.mean(axis=0)
  
      x_train, x_test = model_selection.train_test_split(
          X.astype(np.float32), test_size=(1 / 7.0)
      )
      return x_train, x_test
  
file_path: neupy/examples/autoencoder/variational_autoencoder.py
imports: import numpy
  import tensorflow
  import matplotlib.pyplot
  import matplotlib.gridspec
  from sklearn import model_selection, datasets
  from neupy import algorithms
  from neupy.layers import *
  from neupy.exceptions import LayerConnectionError
label_desc: Use image node_id 1 for calling a global function with example usage: load_data() and returns: x_train,  x_test