code: def run_toy_HMC(gpu_id=None):
      
      X, Y, X_test, Y_test = load_toy()
      minibatch_size = Y.shape[0]
      noise_precision = 1 / 9.0
      net = get_toy_sym(True, noise_precision)
      data_shape = (minibatch_size,) + X.shape[1::]
      data_inputs = {'data': nd.zeros(data_shape, ctx=dev(gpu_id)),
                     'teacher_output_label': nd.zeros((minibatch_size, 1), ctx=dev(gpu_id))}
      initializer = mx.init.Uniform(0.07)
      sample_pool = HMC(net, data_inputs=data_inputs, X=X, Y=Y, X_test=X_test, Y_test=Y_test,
                        sample_num=300000, initializer=initializer, prior_precision=1.0,
                        learning_rate=1E-3, L=10, dev=dev(gpu_id))
docs: Run HMC on toy dataset