tensorflow_probability/examples/vae.py
def build_input_fns(data_dir, batch_size):
  """Builds an Iterator switching between train and heldout data."""

  # Build an iterator over training batches.
  def train_input_fn():
    dataset = static_mnist_dataset(data_dir, "train")
    dataset = dataset.shuffle(50000).repeat().batch(batch_size)
    return tf.compat.v1.data.make_one_shot_iterator(dataset).get_next()

  # Build an iterator over the heldout set.
  def eval_input_fn():
    eval_dataset = static_mnist_dataset(data_dir, "valid")
    eval_dataset = eval_dataset.batch(batch_size)
    return tf.compat.v1.data.make_one_shot_iterator(eval_dataset).get_next()

  return train_input_fn, eval_input_fn
def build_input_fns(data_dir, batch_size):
  """Builds an Iterator switching between train and heldout data."""

  # Build an iterator over training batches.
  def train_input_fn():
    dataset = static_mnist_dataset(data_dir, "train")
    dataset = dataset.shuffle(50000).repeat().batch(batch_size)
    return tf.compat.v1.data.make_one_shot_iterator(dataset).get_next()

  # Build an iterator over the heldout set.
  def eval_input_fn():
    eval_dataset = static_mnist_dataset(data_dir, "valid")
    eval_dataset = eval_dataset.batch(batch_size)
    return tf.compat.v1.data.make_one_shot_iterator(eval_dataset).get_next()

  return train_input_fn, eval_input_fn
def build_input_fns ( data_dir , batch_size ) : # Build an iterator over training batches. def train_input_fn ( ) : dataset = static_mnist_dataset ( data_dir , "train" ) dataset = dataset . shuffle ( 50000 ) . repeat ( ) . batch ( batch_size ) return tf . compat . v1 . data . make_one_shot_iterator ( dataset ) . get_next ( ) # Build an iterator over the heldout set. def eval_input_fn ( ) : eval_dataset = static_mnist_dataset ( data_dir , "valid" ) eval_dataset = eval_dataset . batch ( batch_size ) return tf . compat . v1 . data . make_one_shot_iterator ( eval_dataset ) . get_next ( ) return train_input_fn , eval_input_fn
Builds an Iterator switching between train and heldout data.
Builds an Iterator switching between train and heldout data .