a5dee8148cce0583a9f36779811f1e6a97e18ba3,census/tensorflowcore/trainer/model.py,,input_fn,#Any#Any#Any#Any#Any#,261

Before Change

  if shuffle:
    dataset = dataset.shuffle(buffer_size=batch_size * 10)
  dataset = dataset.repeat(num_epochs)
  dataset = dataset.batch(batch_size)
  iterator = dataset.make_one_shot_iterator()
  features = iterator.get_next()
  return features, features.pop(LABEL_COLUMN)

After Change

  if shuffle:
    dataset = dataset.shuffle(buffer_size=batch_size * 10)
  iterator = dataset.repeat(num_epochs).batch(
      batch_size).make_one_shot_iterator()
  features = iterator.get_next()
  return features, features.pop(LABEL_COLUMN)

Instances