1179e6e7b0caf883b32c4f466208e3373450c1d2,imdb/keras/trainer/model.py,,input_fn,#Any#Any#Any#Any#,64

Before Change

    A tf.estimator.
  
  // Default settings for training
  num_epochs = None
  shuffle = True

  // Override if this is eval
  if mode == tf.estimator.ModeKeys.EVAL:
    num_epochs = 1
    shuffle = False
  y = np.asarray(y).astype("int").reshape((-1, 1))

  return tf.estimator.inputs.numpy_input_fn(
      x,
      y=y,
      batch_size=batch_size,
      num_epochs=num_epochs,
      shuffle=shuffle,
      queue_capacity=50000)


def serving_input_fn():
  Defines the features to be passed to the model during inference.

After Change

    labels = np.asarray(labels).astype("int").reshape((-1, 1))
    inputs = (features, labels)
  // Convert the inputs to a Dataset.
  dataset = tf.data.Dataset.from_tensor_slices(inputs)
  if mode == tf.estimator.ModeKeys.TRAIN:
    dataset = dataset.shuffle(1000).repeat().batch(batch_size)
  if mode == tf.estimator.ModeKeys.EVAL:
    dataset = dataset.batch(batch_size)
  return dataset.make_one_shot_iterator().get_next()


def serving_input_fn():
  Defines the features to be passed to the model during inference.

Instances